source does not extend SegmenterLayer

AllenCell · Dec 10, 2024 · 01204bd · 01204bd
1 parent f2d4cf1
commit 01204bd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/allencell_ml_segmenter/main/segmenter_layer.py b/src/allencell_ml_segmenter/main/segmenter_layer.py
@@ -15,15 +15,15 @@ class ShapesLayer(SegmenterLayer):
 
 
 @dataclass
-class Source(SegmenterLayer):
+class Source:
     path: Optional[Path] = None
 
 
 @dataclass
 class ImageLayer(SegmenterLayer):
     path: Optional[Path] = None
     data: Optional[np.ndarray] = None
-    source: Source = Source(name="sourcename")
+    source: Source = Source()
     metadata: Optional[dict] = None