ensure smrt matrix are float64, not int64

smrt-model · Aug 27, 2019 · 0191203 · 0191203
1 parent bcffc5b
commit 0191203
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/smrt/core/lib.py b/smrt/core/lib.py
@@ -177,7 +177,7 @@ def compress(self, mode=None, auto_reduce_npol=False):
 
 """
         if self.values is 0:
-            return 0
+            return 0.
 
         if self.mtype == "dense5":
             if mode is not None:
@@ -255,7 +255,7 @@ def __abs__(self):
 
     def __getitem__(self, key):
         if self.values is 0:
-            return 0
+            return 0.
         else:
             return self.values[key]
 
@@ -265,7 +265,7 @@ def __setitem__(self, key, v):
     @property
     def diagonal(self):
         if self.values is 0:
-            return np.array([0])
+            return np.array([0.])
         if self.mtype.startswith("diagonal"):
             return self.values
         else: