Adding Imports to __init__ files

minerva/analysis/metrics/__init__.py

@@ -1 +1,3 @@
 from .pixel_accuracy import PixelAccuracy
+
+__all__ = ["PixelAccuracy"]

minerva/data/datasets/__init__.py

@@ -0,0 +1,4 @@
+from .base import SimpleDataset
+from .supervised_dataset import SupervisedReconstructionDataset
+
+__all__ = ["SimpleDataset", "SupervisedReconstructionDataset"]

minerva/data/readers/__init__.py

@@ -0,0 +1,13 @@
+from .patched_array_reader import PatchedArrayReader
+from .png_reader import PNGReader
+from .reader import _Reader
+from .tiff_reader import TiffReader
+from .zarr_reader import PatchedZarrReader
+
+__all__ = [
+    "PatchedArrayReader",
+    "PatchedZarrReader",
+    "PNGReader",
+    "TiffReader",
+    "_Reader",
+]

minerva/models/nets/__init__.py

@@ -0,0 +1,13 @@
+from .base import SimpleSupervisedModel
+from .deeplabv3 import DeepLabV3
+from .setr import SETR_PUP
+from .unet import UNet
+from .wisenet import WiseNet
+
+__all__ = [
+    "SimpleSupervisedModel",
+    "DeepLabV3",
+    "SETR_PUP",
+    "UNet",
+    "WiseNet",
+]

minerva/pipelines/__init__.py

@@ -0,0 +1,3 @@
+from .base import Pipeline
+
+__all__ = ["Pipeline"]

minerva/transforms/__init__.py

@@ -1 +1,3 @@
 from .transform import Flip, PerlinMasker, TransformPipeline, _Transform
+
+__all__ = ["Flip", "PerlinMasker", "TransformPipeline", "_Transform"]

tests/data/datasets/test_simple_dataset.py

@@ -1,6 +1,7 @@
 import numpy as np
 import pytest
-from minerva.data.datasets.base import SimpleDataset
+
+from minerva.data.datasets import SimpleDataset
 
 
 class _SimpleReader:

tests/data/readers/test_patched_array_reader.py

@@ -1,6 +1,6 @@
 import numpy as np
 
-from minerva.data.readers.patched_array_reader import PatchedArrayReader
+from minerva.data.readers import PatchedArrayReader
 
 
 def test_patched_array_reader_no_stride_1():

tests/models/nets/test_setr.py

@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-from minerva.models.nets.setr import SETR_PUP
+from minerva.models.nets import SETR_PUP
 
 
 def test_setr_loss():

tests/models/nets/test_unet.py

@@ -1,8 +1,8 @@
+import lightning as L
 import torch
-
-from minerva.models.nets.unet import UNet
 import torchmetrics
-import lightning as L
+
+from minerva.models.nets import UNet
 from minerva.utils.data import RandomDataModule
 
 

tests/models/nets/test_wisenet.py

@@ -1,6 +1,6 @@
 import torch
 
-from minerva.models.nets.wisenet import WiseNet
+from minerva.models.nets import WiseNet
 
 
 def test_wisenet_loss():