Fix minor regressions

childmindresearch · Nov 2, 2023 · de09c38 · de09c38
1 parent 5897713
commit de09c38
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 5 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "mrirage"
-version = "0.1.0"
+version = "0.1.1"
 authors = ["Florian Rupprecht <[email protected]>"]
 license = "MIT"
 description = "MRI visualization"

diff --git a/src/mrirage/composition/__init__.py b/src/mrirage/composition/__init__.py
@@ -1,6 +1,35 @@
-from .composition import Composition, Style
+from .composition import Composition
 from .grid import CompositionGrid
-from .layer import Layer
+from .layer import (
+    Layer,
+    LayerCoordinate,
+    LayerCross,
+    LayerCrossOrigin,
+    LayerLine,
+    LayerLR,
+    LayerVoxel,
+    LayerVoxelGlass,
+    ColorScale,
+    ColorScaleFromName,
+    ColorScaleSolid,
+    Style,
+)
 from .view import View
 
-__all__ = ["Composition", "CompositionGrid", "Layer", "Style", "View"]
+__all__ = [
+    "Composition",
+    "CompositionGrid",
+    "Style",
+    "View",
+    "Layer",
+    "LayerCoordinate",
+    "LayerCross",
+    "LayerCrossOrigin",
+    "LayerLine",
+    "LayerLR",
+    "LayerVoxel",
+    "LayerVoxelGlass",
+    "ColorScale",
+    "ColorScaleFromName",
+    "ColorScaleSolid",
+]
diff --git a/src/mrirage/composition/view/view.py b/src/mrirage/composition/view/view.py
@@ -29,7 +29,6 @@ def __init__(
         self.axis = axis
 
     def render(self, layers: List[Layer], plt_ax: plt.Axes) -> None:
-        assert self.bounds is not None
         for layer in layers:
             layer.view_render(
                 plt_ax=plt_ax,