ignore errors when dropping columns (#111)

* ignore errors when dropping columns
jerpint · Jun 26, 2023 · 6c956ff · 6c956ff
1 parent e4d8b7a
commit 6c956ff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/buster/completers/base.py b/buster/completers/base.py
@@ -88,7 +88,7 @@ def to_json(self, columns_to_ignore: Optional[list[str]] = None) -> Any:
 
         def encode_df(df: pd.DataFrame) -> dict:
             if columns_to_ignore is not None:
-                df = df.drop(columns=columns_to_ignore)
+                df = df.drop(columns=columns_to_ignore, errors="ignore")
             return df.to_json(orient="index")
 
         custom_encoder = {