Skip to content

Commit 3bf8b1f

Browse files
author
Rohan Yadav
committed
numpy: add an allowlist for FROSTT tensors that run successfully
I have to try some more tensors, but the very large ones like amazon-reviews, patents or reddit don't succesfully load.
1 parent f86d264 commit 3bf8b1f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

numpy/util.py

+2
Original file line numberDiff line numberDiff line change
@@ -157,6 +157,8 @@ def __init__(self):
157157
data = os.path.join(TENSOR_PATH, "FROSTT")
158158
frostttensors = glob.glob(os.path.join(data, "*.tns"))
159159
self.tensors = [FROSTTTensor(t) for t in frostttensors]
160+
allowlist = ["nips", "uber-pickups", "chicago-crime", "lbnl-network", "enron", "nell-2", "vast"]
161+
self.tensors = [t for t in self.tensors if str(t) in allowlist]
160162

161163
def getTensors(self):
162164
return self.tensors

0 commit comments

Comments
 (0)