diff --git a/auto_trainer/test_auto_trainer.py b/auto_trainer/test_auto_trainer.py index 9a1ff554..c8953f43 100644 --- a/auto_trainer/test_auto_trainer.py +++ b/auto_trainer/test_auto_trainer.py @@ -78,7 +78,7 @@ def _assert_train_handler(train_run): @pytest.mark.parametrize("model", MODELS) -def test_train(model: Tuple[str, str]): +def test_train(model: Tuple[str, str]): # dataset, label_columns = _get_dataset(model[1]) is_test_passed = True