feat(inline): disable load_data()

crunchdao · Oct 25, 2024 · ce87a12 · ce87a12
1 parent 0a68b08
commit ce87a12
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/crunch/inline.py b/crunch/inline.py
@@ -34,6 +34,10 @@ def _competition(self):
         return competition
 
     def load_data(self, **kwargs) -> typing.Tuple[LoadedData, LoadedData, LoadedData]:
+        if self._competition.format == api.CompetitionFormat.SPATIAL:
+            logging.error(f"Please follow the competition instructions to load the data.")
+            return None, None, None
+
         if self._competition.format == api.CompetitionFormat.STREAM:
             logging.error(f"Please call `.load_streams()` instead.")
             return None, None, None
@@ -64,6 +68,10 @@ def load_data(self, **kwargs) -> typing.Tuple[LoadedData, LoadedData, LoadedData
         return x_train, y_train, x_test
 
     def load_streams(self, **kwargs) -> typing.Tuple[Streams, Streams]:
+        if self._competition.format == api.CompetitionFormat.SPATIAL:
+            logging.error(f"Please follow the competition instructions to load the data.")
+            return None, None
+
         if self._competition.format != api.CompetitionFormat.STREAM:
             logging.error(f"Please call `.load_data()` instead.")
             return None, None