LLCAXCHZF-61/use drop with errors ignore

DataShades · Oct 18, 2024 · 833ed81 · 833ed81
1 parent d0b9191
commit 833ed81
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/ckanext/charts/fetchers.py b/ckanext/charts/fetchers.py
@@ -71,11 +71,7 @@ def fetch_data(self) -> pd.DataFrame:
                 .select_from(sa.table(self.resource_id))
                 .limit(self.limit),
                 get_read_engine(),
-            )
-
-            existing_columns = {col for col in {'_id', '_full_text'} if col in df.columns}
-            if existing_columns:
-                df = df.drop(columns=existing_columns)
+            ).drop(columns=["_id", "_full_text"], errors='ignore')
 
             if "date_time" in df.columns:
                 try: