fixed minor import error

kmheckel · Feb 8, 2024 · 146dc35 · 146dc35
1 parent 94c4c11
commit 146dc35
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/spyx/loaders.py b/spyx/loaders.py
@@ -11,6 +11,8 @@
     import torchvision as tv
     from torch.utils.data import DataLoader, Subset
     from sklearn.model_selection import train_test_split
+
+    from .data import rate_code
 except ImportError:
     optional_dependencies_installed = False