Update generator.py

globus-labs · Oct 18, 2023 · 032e64f · 032e64f
1 parent 90727aa
commit 032e64f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mofa/generator.py b/mofa/generator.py
@@ -10,7 +10,7 @@
 import os
 
 def train_generator
-        config: str="config.yaml"
+        config: str="config.yaml",
         # starting_model: str | Path,
         examples: str|list[MOFRecord]="../argonne_gnn_gitlab/DiffLinker/data/geom/datasets",
         num_epochs: int=10