From 0c7b93b9a193275e4b6fdf8594f3eebe160958d5 Mon Sep 17 00:00:00 2001 From: Niko Yasui Date: Tue, 5 Nov 2024 15:00:06 -0700 Subject: [PATCH] refactor(test): remove hardcoded part name in acceptance test --- tests/acceptance/my_experiment.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/tests/acceptance/my_experiment.py b/tests/acceptance/my_experiment.py index 5588eb7..1dd0a5f 100644 --- a/tests/acceptance/my_experiment.py +++ b/tests/acceptance/my_experiment.py @@ -34,15 +34,11 @@ def run(self) -> str: def main(): cmdline = parser.parse_args() - # make sure we are using softmaxAC - if cmdline.part != "softmaxAC": - raise ValueError(f"Unknown part: {cmdline.part}") - # do some rng control random.seed(cmdline.seed) # extract configs from the database - exp = ExperimentDefinition("softmaxAC", cmdline.version) + exp = ExperimentDefinition(cmdline.part, cmdline.version) # TODO: don't overwrite this exp.get_results_path = lambda *args, **kwargs: cmdline.results_path # overwrite results path config = exp.get_config(cmdline.config_id)