diff --git a/reprodExperiment.py b/main.py similarity index 81% rename from reprodExperiment.py rename to main.py index c3bcbef83afade66583c8530488fb5212632ea04..c89dda302b5e59c2527b1561fa62b90285b87fff 100644 --- a/reprodExperiment.py +++ b/main.py @@ -16,4 +16,7 @@ if (__name__ == "__main__"): if not (args.branch): print("Please specify a branch") exit(1) - loadExperiment.run(args.load, args.branch) \ No newline at end of file + loadExperiment.run(args.load, args.branch) + if (not args.save) and (not args.load): + print("Please specify an action") + exit(1) \ No newline at end of file diff --git a/registerExperiment.py b/registerExperiment.py index aacb0dcaa0e798ed45c2c43613ad8ccf032e97c6..5fbe12f0f60e5171e546c5604bc109aaff0edf11 100644 --- a/registerExperiment.py +++ b/registerExperiment.py @@ -113,14 +113,17 @@ def branchExists(branchName) -> bool: return branchName in repository.references def pushBranch(version=1) -> None: + print("Pushing branch...") while (branchExists(f"{experimentName}Experiment{version}")): version += 1 else: + print(f"creating {experimentName}Experiment{version} branch and pushing changes to it ...") repository.git.checkout('-b', f"{experimentName}Experiment{version}") repository.git.add(all=True) repository.git.commit(m=f"{experimentName}Experiment{version}") repository.git.push('--set-upstream', repository.remote().name, f"{experimentName}Experiment{version}") repository.git.checkout(experimentName) + print("done") def genChecksum(file) -> str : hash_md5 = hashlib.md5()