Fixing mypy errors in optika.transforms.

sun-data · Aug 14, 2023 · 1a0fee5 · 1a0fee5
1 parent 3a4211b
commit 1a0fee5
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/optika/transforms.py b/optika/transforms.py
@@ -77,7 +77,9 @@ class Translation(AbstractTransform):
     A translation-only vector transformation.
     """
 
-    displacement: na.Cartesian3dVectorArray = dataclasses.MISSING
+    displacement: na.Cartesian3dVectorArray = dataclasses.field(
+        default_factory=na.Cartesian3dVectorArray
+    )
     """The magnitude of the translation along each axis"""
 
     @property
@@ -157,7 +159,7 @@ def transforms(self) -> Iterator[AbstractTransform]:
             return iter(self)
 
     @property
-    def matrix(self) -> na.Cartesian3dMatrixArray:
+    def matrix(self) -> na.AbstractCartesian3dMatrixArray:
         rotation = na.Cartesian3dIdentityMatrixArray()
 
         for transform in reversed(list(self.transforms)):
@@ -169,7 +171,7 @@ def matrix(self) -> na.Cartesian3dMatrixArray:
     @property
     def vector(self) -> na.Cartesian3dVectorArray:
         rotation = na.Cartesian3dIdentityMatrixArray()
-        translation = 0
+        translation = na.Cartesian3dVectorArray()
 
         for transform in reversed(list(self.transforms)):
             if transform is not None: