Skip to content

Commit 74988ac

Browse files
committed
remove file dependency from xcebra int test
1 parent 61cb9b7 commit 74988ac

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

tests/test_integration_xcebra.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,16 @@
1616

1717
@pytest.fixture
1818
def synthetic_data():
19-
with open('examples/synthetic_data.pkl', 'rb') as file:
19+
import os
20+
import urllib.request
21+
22+
url = "https://cebra.fra1.digitaloceanspaces.com/xcebra_synthetic_data.pkl"
23+
filepath = "/tmp/synthetic_data.pkl"
24+
25+
if not os.path.exists(filepath):
26+
urllib.request.urlretrieve(url, filepath)
27+
28+
with open(filepath, 'rb') as file:
2029
return pickle.load(file)
2130

2231

0 commit comments

Comments
 (0)