From a7661fd8702ddb07fef324cb295d9b31a9c521ac Mon Sep 17 00:00:00 2001
From: Jeremy Pinto <jerpint@gmail.com>
Date: Mon, 26 Jun 2023 15:07:45 -0400
Subject: [PATCH] ignore errors when dropping columns

---
 buster/completers/base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/buster/completers/base.py b/buster/completers/base.py
index 14071d7..0d624cd 100644
--- 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 = {