trying to fix some serialization issues

when using a remote catalog I'm getting weird errors like KeyError on 'dtype' and cannot serialize a Series
ContinuumIO · Oct 25, 2021 · 4ff6eec · 4ff6eec
1 parent 42114ad
commit 4ff6eec
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/intake_metabase/source.py b/intake_metabase/source.py
@@ -95,7 +95,7 @@ def _get_schema(self):
             self._df = self._metabase.get_card(self.question)
 
         return Schema(datashape=None,
-                      dtype=self._df.dtypes,
+                      dtype={n: str(t) for (n, t) in self._df.dtypes.items()},
                       shape=(None, len(self._df.columns)),
                       npartitions=1,
                       extra_metadata={})
@@ -112,7 +112,7 @@ def to_dask(self):
         raise NotImplementedError()
 
     def _close(self):
-        self._dataframe = None
+        self._df = None
 
 
 class MetabaseTableSource(DataSource):
@@ -140,7 +140,7 @@ def _get_schema(self):
             self._df = self._metabase.get_table(self.database, self.table, self.query)
 
         return Schema(datashape=None,
-                      dtype=self._df.dtypes,
+                      dtype={n: str(t) for (n, t) in self._df.dtypes.items()},
                       shape=(None, len(self._df.columns)),
                       npartitions=1,
                       extra_metadata={})
@@ -157,7 +157,7 @@ def to_dask(self):
         raise NotImplementedError()
 
     def _close(self):
-        self._dataframe = None
+        self._df = None
 
 
 class MetabaseAPI():