fix: torch20 cleanup (#81)

* 🗑️ remove redundant method * 🎨 formatting * 🔥 py3 super() and remove unused methods * ✅ add tests for transformer * 🎨 formatting * ✅ add tests for tower unet * 🎨 formatting * 🔥 remove FOJ * 🔥 py3 super() and remove unused methods * 🎨 formatting * 🔥 py3 super() and remove unused methods
jgrss · Jul 31, 2024 · 42f90e5 · 42f90e5
1 parent 0e966be
commit 42f90e5
Show file tree

Hide file tree

Showing 27 changed files with 355 additions and 4,546 deletions.
diff --git a/src/cultionet/data/datasets.py b/src/cultionet/data/datasets.py
@@ -20,7 +20,7 @@
 from ..augment.augmenters import Augmenters
 from ..errors import TensorShapeError
 from ..utils.logging import set_color_logger
-from ..utils.model_preprocessing import TqdmParallel
+from ..utils.model_preprocessing import ParallelProgress
 from ..utils.normalize import NormValues
 from .data import Data
 from .spatial_dataset import SpatialDataset
@@ -257,7 +257,7 @@ def check_dims(
             backend="loky",
             n_jobs=self.processes,
         ):
-            with TqdmParallel(
+            with ParallelProgress(
                 tqdm_kwargs={
                     "total": len(self),
                     "desc": "Checking dimensions",