[FEATURE] - Rds 1145/add manual action trigger for test harness

Co-authored-by: Andre Manoel <[email protected]> Co-authored-by: Kendrick Boyd <[email protected]> GitOrigin-RevId: 81a857fa84dafacf3fb4e036c9ed74cc07724733
gretelai · Dec 17, 2024 · d1d2beb · d1d2beb
1 parent bb523e9
commit d1d2beb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gretel_trainer/benchmark/gretel/models.py b/src/gretel_trainer/benchmark/gretel/models.py
@@ -8,7 +8,7 @@
 
 from gretel_client.projects.exceptions import ModelConfigError
 from gretel_client.projects.models import read_model_config
-from gretel_trainer.benchmark.core import BenchmarkException, Dataset, Datatype
+from gretel_trainer.benchmark.core import BenchmarkException, Dataset
 
 GretelModelConfig = Union[str, Path, dict]