From d1d2beb2a2676688de4e6e77563d992a6eaae788 Mon Sep 17 00:00:00 2001 From: Aaron Gonzales Date: Mon, 16 Dec 2024 19:05:22 -0700 Subject: [PATCH] [FEATURE] - Rds 1145/add manual action trigger for test harness Co-authored-by: Andre Manoel Co-authored-by: Kendrick Boyd GitOrigin-RevId: 81a857fa84dafacf3fb4e036c9ed74cc07724733 --- src/gretel_trainer/benchmark/gretel/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/gretel_trainer/benchmark/gretel/models.py b/src/gretel_trainer/benchmark/gretel/models.py index d627c602..324860d0 100644 --- 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]