Cap progress to 0.99 if experiment is still running (#861)

* Cap progress to 0.99 if experiment is still running
h2oai · Sep 19, 2024 · 32fa36b · 32fa36b
1 parent 3284a5c
commit 32fa36b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/llm_studio/app_utils/utils.py b/llm_studio/app_utils/utils.py
@@ -1850,6 +1850,13 @@ def get_experiments(
         # make sure progress is 100% for finished experiments
         df.loc[df.status == "finished", "progress"] = "1.0"
 
+        # make sure that if experiment is running the progress is at most 99%
+        df.loc[
+            (df.status == "running")
+            & (pd.to_numeric(df.progress, errors="coerce") > 0.99),
+            "progress",
+        ] = ".99"
+
         df["info"] = np.where(
             (df["status"] == "running") & (df["eta"] != ""),
             df["eta"].apply(lambda x: f"ETA: {x}"),