From 032e64f9336b3e77c5bce28b085a9510c271e4e6 Mon Sep 17 00:00:00 2001
From: hyunp2 <42776897+hyunp2@users.noreply.github.com>
Date: Wed, 18 Oct 2023 00:47:57 -0500
Subject: [PATCH] Update generator.py

---
 mofa/generator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mofa/generator.py b/mofa/generator.py
index 1b88cbae..4dc5d2f0 100644
--- 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