skipping unavailable metrics

eflows4hpc · Feb 5, 2024 · d9110c7 · d9110c7
1 parent e1d517b
commit d9110c7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/model_search_upload.py b/model_search_upload.py
@@ -92,6 +92,9 @@ def uploat_to_mlflow(temp_dir, **context):
                     mlflow.log_param(key=parname, value=parvalue)
 
                 for m in metrics:
+                    if m not in dct:
+                        continue
+
                     print(f"Logging metric {m} {dct[m][i]}")
                     mlflow.log_metric(key=m, value=dct[m][i])