From 4ff6eecb4d1fd99bbf083e1fb49c0af7a5adaf52 Mon Sep 17 00:00:00 2001
From: Albert DeFusco <albert.defusco@me.com>
Date: Mon, 25 Oct 2021 14:58:58 -0500
Subject: [PATCH] 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
---
 intake_metabase/source.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/intake_metabase/source.py b/intake_metabase/source.py
index 0608380..28ddf30 100644
--- 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():