fix(pdf): return - if metric not found

crunchdao · Dec 20, 2023 · a7f7cea · a7f7cea
1 parent 3dac3a3
commit a7f7cea
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/backtest/export/pdf.py b/backtest/export/pdf.py
@@ -115,6 +115,10 @@ def get_metric(column: str, name: str):
                 except:
                     pass
 
+                if name not in df_metrics.index:
+                    print(f"[warning] invalid metric name: `{name}`: not in index: {df_metrics.index}")
+                    return "-"
+
                 value = df_metrics.loc[name, column]
                 if name in NOT_PERCENT:
                     return value