Fix: Git record difference check

ecmwf · Sep 12, 2024 · d34f946 · d34f946
1 parent 340842c
commit d34f946
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/anemoi/inference/checkpoint/__init__.py b/src/anemoi/inference/checkpoint/__init__.py
@@ -118,6 +118,10 @@ def validate_environment(
                 )
 
         for git_record in train_environment["git_versions"].keys():
+            file_record = train_environment["git_versions"][git_record]["git"]
+            if file_record["modified_files"] == 0 and file_record["untracked_files"] == 0:
+                continue
+
             if git_record not in inference_environment["git_versions"]:
                 invalid_messages["uncommitted"].append(
                     f"Training environment contained uncommitted change missing in inference environment: {git_record}"
@@ -131,6 +135,10 @@ def validate_environment(
                 )
 
         for git_record in inference_environment["git_versions"].keys():
+            file_record = inference_environment["git_versions"][git_record]["git"]
+            if file_record["modified_files"] == 0 and file_record["untracked_files"] == 0:
+                continue
+
             if git_record not in train_environment["git_versions"]:
                 invalid_messages["uncommitted"].append(
                     f"Inference environment contains uncommited changes missing in training: {git_record}"