diff --git a/registerExperiment.py b/registerExperiment.py index aa1b061659e216b40d08a6c9fbaec640f49e35ec..e763be809d7765962b4e3adb4183826215846739 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: