Skip to content
Snippets Groups Projects
Commit 2fa33f20 authored by AxelCarayon's avatar AxelCarayon
Browse files

ne crash plus si le fichier experimentResume.yaml n'existe pas

parent 6c741966
No related branches found
No related tags found
No related merge requests found
...@@ -209,19 +209,23 @@ def checkGeneratedFiles() -> None : ...@@ -209,19 +209,23 @@ def checkGeneratedFiles() -> None :
def writeInYaml() -> None: def writeInYaml() -> None:
with open(EXPERIMENT_RESUME, "r") as yamlFile: if fileExists(EXPERIMENT_RESUME):
cur_yaml = yaml.safe_load(yamlFile) with open(EXPERIMENT_RESUME, "r") as yamlFile:
cur_yaml.update({"name":experimentName}) cur_yaml = yaml.safe_load(yamlFile)
cur_yaml.update({"commands":commandsFile}) cur_yaml.update({"name":experimentName})
cur_yaml.update({"inputs":inputFiles}) cur_yaml.update({"commands":commandsFile})
cur_yaml.update({"outputs":outputFiles}) cur_yaml.update({"inputs":inputFiles})
cur_yaml.update({"params":paramsFiles}) cur_yaml.update({"outputs":outputFiles})
cur_yaml.update({"instruction":instructionFile}) cur_yaml.update({"params":paramsFiles})
cur_yaml.update({"dockerfile":fileExists(DOCKERFILE)}) cur_yaml.update({"instruction":instructionFile})
checksums = {"checksums":genChecksums()} cur_yaml.update({"dockerfile":fileExists(DOCKERFILE)})
cur_yaml.update(checksums) checksums = {"checksums":genChecksums()}
with open(EXPERIMENT_RESUME, 'w') as yamlFile: cur_yaml.update(checksums)
yaml.safe_dump(cur_yaml, yamlFile) 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: def pushBranch(version=1) -> None:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment