Fixed a bug in MultilayerFilm.shape and MultilayerMirror.shape wh…

…ere a `ValueError` was raised if `layers` was not a sequence.
sun-data · Nov 6, 2024 · b184e5c · b184e5c
1 parent 5f841c6
commit b184e5c
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/optika/materials/_multilayers.py b/optika/materials/_multilayers.py
@@ -883,9 +883,13 @@ class MultilayerFilm(
 
     @property
     def shape(self) -> dict[str, int]:
-        return na.broadcast_shapes(
-            *[optika.shape(layer) for layer in self.layers],
-        )
+        try:
+            shape_layer = na.broadcast_shapes(
+                *[optika.shape(layer) for layer in self.layers],
+            )
+        except TypeError:
+            shape_layer = optika.shape(self.layers)
+        return shape_layer
 
 
 @dataclasses.dataclass(eq=False, repr=False)
@@ -1099,7 +1103,13 @@ class MultilayerMirror(
 
     @property
     def shape(self) -> dict[str, int]:
+        try:
+            shape_layer = na.broadcast_shapes(
+                *[optika.shape(layer) for layer in self.layers],
+            )
+        except TypeError:
+            shape_layer = optika.shape(self.layers)
         return na.broadcast_shapes(
-            *[optika.shape(layer) for layer in self.layers],
+            shape_layer,
             optika.shape(self.substrate),
         )