fix

pfnet · Dec 17, 2020 · b2f46c0 · b2f46c0
1 parent 51c0127
commit b2f46c0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pfrl/experiments/prepare_output_dir.py b/pfrl/experiments/prepare_output_dir.py
@@ -16,7 +16,7 @@ def is_under_git_control():
     return pfrl.utils.is_return_code_zero(["git", "rev-parse"])
 
 
-def generate_exp_id(prefix=None) -> str:
+def generate_exp_id(prefix=None, argv=sys.argv) -> str:
     """Generate reproducible, unique and deterministic experiment id
 
     The generated id will be string generated from prefix, Git
@@ -39,7 +39,7 @@ def generate_exp_id(prefix=None) -> str:
     names.append(head.decode())
 
     # Caveat: does not work with new files that are not yet cached
-    sources = [subprocess.check_output("git diff HEAD".split()), pickle.dumps(sys.argv)]
+    sources = [subprocess.check_output("git diff HEAD".split()), pickle.dumps(argv)]
 
     for source in sources:
         names.append("%08x" % crc32(source))