Skip to content

Commit 58f6052

Browse files
authored
Merge pull request #158 from Sabine143/main
Solve diversity_filter issue (diversity_filter not applied even though present in the config file) by changing “if "type" in config:” into “if memory_type:”
2 parents e2befbf + 1edfb90 commit 58f6052

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

reinvent/runmodes/RL/run_staged_learning.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def setup_diversity_filter(config: SectionDiversityFilter, rdkit_smiles_flags: d
4545

4646
memory_type = config.type
4747

48-
if "type" in config:
48+
if hasattr(config, "type"):
4949
diversity_filter = getattr(memories, memory_type)
5050
else:
5151
return None

0 commit comments

Comments
 (0)