remove print and added comment

mlcommons · Nov 3, 2023 · b97fb05 · b97fb05
1 parent 39b2428
commit b97fb05
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/GANDLF/utils/modelio.py b/GANDLF/utils/modelio.py
@@ -155,14 +155,14 @@ def save_model(
     model_dict["parameters"] = params
 
     try:
-        print("DEBUG os.getcwd():", os.getcwd())
+        # this will try to encode the git hash of the current GaNDLF codebase, and reverts to "None" if not found
         model_dict["git_hash"] = (
             subprocess.check_output(["git", "rev-parse", "HEAD"], cwd=os.getcwd())
             .decode("ascii")
             .strip()
         )
     except (subprocess.CalledProcessError, FileNotFoundError):
-        model_dict["git_hash"] = None
+        model_dict["git_hash"] = "None"
 
     torch.save(model_dict, path)