Make this more robust

Because bond type does not return currency. See also alvarobartt/investpy#572
ymyke · May 23, 2022 · eb514cf · eb514cf
1 parent dfb418a
commit eb514cf
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tessa/investing.py b/tessa/investing.py
@@ -26,7 +26,10 @@
 
 def get_currency_from_dataframe(df: pd.DataFrame) -> str:
     """Get currency from investpy dataframe."""
-    currencies = list(df["Currency"].unique())
+    try:
+        currencies = list(df["Currency"].unique())
+    except KeyError:
+        return "UNKNOWN"
     if len(currencies) > 1:
         raise ValueError(f"Expected only one currency, got {currencies}.")
     return currencies[0]