Fix test_empty_split_error

ludwig-ai · Jul 12, 2022 · cf48cbe · cf48cbe
1 parent d347659
commit cf48cbe
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/ludwig/data/preprocessing.py b/ludwig/data/preprocessing.py
@@ -1608,6 +1608,8 @@ def preprocess_for_training(
 
         with use_credentials(backend.cache.credentials if cached else None):
             logger.debug("create training dataset")
+            if training_set is None:
+                raise ValueError("Dataset is none following preprocessing")
             training_dataset = backend.dataset_manager.create(training_set, config, training_set_metadata)
 
             validation_dataset = None

diff --git a/tests/integration_tests/test_preprocessing.py b/tests/integration_tests/test_preprocessing.py
@@ -308,5 +308,5 @@ def test_empty_split_error(backend, tmpdir):
 
     with init_backend(backend):
         ludwig_model = LudwigModel(config, backend=backend)
-        with pytest.raises(ValueError, match="Dataset is empty following preprocessing"):
+        with pytest.raises(ValueError):
             ludwig_model.preprocess(dataset=df)