Update util.py to pass codestyle

weiya711 · Aug 8, 2023 · 11502cb · 11502cb
1 parent 6aa9f6f
commit 11502cb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sam/util.py b/sam/util.py
@@ -142,7 +142,7 @@ def load(self, path):
 
 
 # PydataSparseTensorLoader loads a sparse tensor from a file into
-# a pydata.sparse tensor.
+# a sparse tensor.
 # class PydataSparseTensorLoader:
 #     def __init__(self):
 #         self.loader = TnsFileLoader()
@@ -547,7 +547,7 @@ def load(self, tensor, suiteSparse, cast, format_str):
 
 
 # PydataMatrixMarketTensorLoader loads tensors in the matrix market format
-# into pydata.sparse matrices.
+# into sparse matrices.
 # class PydataMatrixMarketTensorLoader:
 #     def __init__(self):
 #         pass
@@ -601,7 +601,7 @@ def safeCastPydataTensorToInts(tensor):
         # else:
         #     data[i] = int(tensor.data[i])
         data[i] = round_sparse(tensor.data[i])
-    return pydata.sparse.COO(tensor.coords, data, tensor.shape)
+    return sparse.COO(tensor.coords, data, tensor.shape)
 
 
 def parse_taco_format(infilename, outdir, tensorname, format_str):