Small fix

greydanus · May 6, 2024 · 01ea44f · 01ea44f
1 parent 011d9be
commit 01ea44f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mnist1d/data.py b/mnist1d/data.py
@@ -4,6 +4,7 @@
 import numpy as np
 import os
 import requests
+import mnist1d
 from mnist1d.transform import transform
 from mnist1d.utils import from_pickle, to_pickle, ObjectView, set_seed