Fix issue where plot data was always stored in db (#599)

datalab-org · Feb 18, 2024 · 1fa1480 · 1fa1480
1 parent 74d64bc
commit 1fa1480
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pydatalab/pydatalab/blocks/base.py b/pydatalab/pydatalab/blocks/base.py
@@ -102,15 +102,17 @@ def __init__(
     def to_db(self):
         """returns a dictionary with the data for this
         block, ready to be input into mongodb"""
+
         LOGGER.debug("Casting block %s to database object.", self.__class__.__name__)
 
+        if "bokeh_plot_data" in self.data:
+            self.data.pop("bokeh_plot_data")
+
         if "file_id" in self.data:
             dict_for_db = self.data.copy()  # gross, I know
             dict_for_db["file_id"] = ObjectId(dict_for_db["file_id"])
             return dict_for_db
 
-        if "bokeh_plot_data" in self.data:
-            self.data.pop("bokeh_plot_data")
         return self.data
 
     @classmethod