diff --git a/clemgame/clemgame.py b/clemgame/clemgame.py index 10b11d7207..52e7972ad7 100644 --- a/clemgame/clemgame.py +++ b/clemgame/clemgame.py @@ -838,12 +838,9 @@ def add_game_instance(self, experiment: Dict, game_id): def on_generate(self): raise NotImplementedError() - def generate(self): + def generate(self, filename="instances.json"): self.on_generate() - self.store() - - def store(self): - self.store_file(self.instances, "instances.json", sub_dir="in") + self.store_file(self.instances, filename, sub_dir="in") def load_benchmarks(do_setup: bool = True) -> List[GameBenchmark]: