added unit to dot

Author: christinahedges

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "sparse3d"
-version = "1.2.5"
+version = "1.2.6"
 description = ""
 authors = ["Christina Hedges <christina.l.hedges@nasa.gov>"]
 packages = [{include = "sparse3d", from = "src"}]

src/sparse3d/sparse3d.py

@@ -260,6 +260,8 @@ def dot(self, other: np.ndarray) -> np.ndarray:
                 .reshape((*self.imshape, nt))
                 .transpose([2, 0, 1])
             )
+        if hasattr(other, "unit"):
+            return result * other.unit
         return result
 
     def _index(self, offset=(0, 0)):