Fixed missing_fraction and sparsity_fraction error associated with te…

…nsorly backend
earmingol · Dec 5, 2022 · feff9b6 · feff9b6
1 parent 6915c5f
commit feff9b6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cell2cell/tensor/tensor.py b/cell2cell/tensor/tensor.py
@@ -808,6 +808,8 @@ def __init__(self, rnaseq_matrices, ppi_data, order_labels=None, context_names=N
         self.outer_fraction = outer_fraction
         if device is None:
             self.tensor = tl.tensor(tensor)
+            self.loc_nans = tl.tensor(self.loc_nans)
+            self.loc_zeros = tl.tensor(self.loc_zeros)
             self.mask = mask
         else:
             if tl.get_backend() == 'pytorch': # Potential TODO: Include other backends that support different devices