Skip to content

Commit

Permalink
refactor(test): remove hardcoded part name in acceptance test
Browse files Browse the repository at this point in the history
  • Loading branch information
yasuiniko committed Nov 5, 2024
1 parent a95e102 commit 0c7b93b
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions tests/acceptance/my_experiment.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 0c7b93b

Please sign in to comment.