Platma commited on
Commit
4bf9627
·
verified ·
1 Parent(s): 0d91645

Update src/main.py

Browse files
Files changed (1) hide show
  1. src/main.py +1 -1
src/main.py CHANGED
@@ -58,7 +58,7 @@ def schedule_retrain(payload: WebhookPayload):
58
  list_doc = yaml.safe_load(f)
59
  list_doc['project_name'] = datetime.datetime.now().isoformat()
60
 
61
- with open(yaml_path) as f:
62
  yaml.dump(list_doc, f, default_flow_style=False)
63
 
64
  result = Popen(['autotrain', '--config', yaml_path]) # project = AutoTrain.create_project(payload)
 
58
  list_doc = yaml.safe_load(f)
59
  list_doc['project_name'] = datetime.datetime.now().isoformat()
60
 
61
+ with open(yaml_path, "w") as f:
62
  yaml.dump(list_doc, f, default_flow_style=False)
63
 
64
  result = Popen(['autotrain', '--config', yaml_path]) # project = AutoTrain.create_project(payload)