WayScience · wli51 · Dec 10, 2025 · Nov 20, 2025 · Nov 20, 2025 · Nov 20, 2025
diff --git a/examples/1.modular_unet_example.ipynb b/examples/1.modular_unet_example.ipynb
diff --git a/examples/nbconverted/1.modular_unet_example.py b/examples/nbconverted/1.modular_unet_example.py
diff --git a/src/virtual_stain_flow/models/blocks/__init__.py b/src/virtual_stain_flow/models/blocks/__init__.py
@@ -0,0 +1,34 @@
+from .blocks import (
+    AbstractBlock,
+    Conv2DConvNeXtBlock,
+    Conv2DNormActBlock,
+)
+from .up_down_blocks import (
+	Bilinear2DUpsampleBlock,
+	Conv2DDownBlock,
+	ConvTrans2DUpBlock,
+	IdentityBlock,
+	MaxPool2DDownBlock,
+	PixelShuffle2DUpBlock,
+)
+from .utils import (
+    ActivationType,
+    get_activation,
+    get_norm
+)
+
+
+__all__ = [
+    "AbstractBlock",
+    "Conv2DConvNeXtBlock",
+    "Conv2DNormActBlock",
+    "Bilinear2DUpsampleBlock",
+    "Conv2DDownBlock",
+    "ConvTrans2DUpBlock",
+    "IdentityBlock",
+    "MaxPool2DDownBlock",
+    "PixelShuffle2DUpBlock",
+    "ActivationType",
+    "get_activation",
+    "get_norm",
+]
diff --git a/src/virtual_stain_flow/models/blocks.py → ...irtual_stain_flow/models/blocks/blocks.py b/src/virtual_stain_flow/models/blocks.py → ...irtual_stain_flow/models/blocks/blocks.py