diff --git a/sam/util.py b/sam/util.py index 67d3fcf4..0658de93 100644 --- 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):