From 2fa33f2053fdb37738e2b3d8f0b413b789459646 Mon Sep 17 00:00:00 2001
From: AxelCarayon <axel.carayon@gmail.com>
Date: Tue, 1 Mar 2022 17:52:56 +0100
Subject: [PATCH] ne crash plus si le fichier experimentResume.yaml n'existe
pas
---
registerExperiment.py | 30 +++++++++++++++++-------------
1 file changed, 17 insertions(+), 13 deletions(-)
diff --git a/registerExperiment.py b/registerExperiment.py
index aa1b061..e763be8 100644
--- a/registerExperiment.py
+++ b/registerExperiment.py
@@ -209,19 +209,23 @@ def checkGeneratedFiles() -> None :
def writeInYaml() -> None:
- with open(EXPERIMENT_RESUME, "r") as yamlFile:
- cur_yaml = yaml.safe_load(yamlFile)
- cur_yaml.update({"name":experimentName})
- cur_yaml.update({"commands":commandsFile})
- cur_yaml.update({"inputs":inputFiles})
- cur_yaml.update({"outputs":outputFiles})
- cur_yaml.update({"params":paramsFiles})
- cur_yaml.update({"instruction":instructionFile})
- cur_yaml.update({"dockerfile":fileExists(DOCKERFILE)})
- checksums = {"checksums":genChecksums()}
- cur_yaml.update(checksums)
- with open(EXPERIMENT_RESUME, 'w') as yamlFile:
- yaml.safe_dump(cur_yaml, yamlFile)
+ if fileExists(EXPERIMENT_RESUME):
+ with open(EXPERIMENT_RESUME, "r") as yamlFile:
+ cur_yaml = yaml.safe_load(yamlFile)
+ cur_yaml.update({"name":experimentName})
+ cur_yaml.update({"commands":commandsFile})
+ cur_yaml.update({"inputs":inputFiles})
+ cur_yaml.update({"outputs":outputFiles})
+ cur_yaml.update({"params":paramsFiles})
+ cur_yaml.update({"instruction":instructionFile})
+ cur_yaml.update({"dockerfile":fileExists(DOCKERFILE)})
+ checksums = {"checksums":genChecksums()}
+ cur_yaml.update(checksums)
+ with open(EXPERIMENT_RESUME, 'w') as yamlFile:
+ yaml.safe_dump(cur_yaml, yamlFile)
+ else:
+ with open(EXPERIMENT_RESUME, "w") as yamlFile:
+ yaml.safe_dump({"name":experimentName, "commands":commandsFile, "inputs":inputFiles, "outputs":outputFiles, "params":paramsFiles, "instruction":instructionFile, "dockerfile":fileExists(DOCKERFILE), "checksums":genChecksums()}, yamlFile)
def pushBranch(version=1) -> None:
--
GitLab