diff --git a/crunch/inline.py b/crunch/inline.py index a6d91ee..9cfa27d 100644 --- 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