only quant

lux-org · Mar 31, 2021 · e32f8ae · e32f8ae
1 parent f063bf8
commit e32f8ae
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lux/core/frame.py b/lux/core/frame.py
@@ -498,12 +498,14 @@ def maintain_recs(self, is_series="DataFrame"):
                 if len(vlist) > 0:
                     rec_df._recommendation[action_type] = vlist
             rec_df._rec_info = rec_infolist
-            if len(self.columns) == 2 or len(self.columns) == 3:
+            quantitative_columns = [i for i in self.dtypes if i != 'O' and i != 'str']
+            if len(quantitative_columns) == 2 or len(quantitative_columns) == 3:
                 self.current_vis = VisList([i for i in self.columns], self)
             self._widget = rec_df.render_widget()
         # re-render widget for the current dataframe if previous rec is not recomputed
         elif show_prev:
-            if len(self.columns) == 2 or len(self.columns) == 3:
+            quantitative_columns = [i for i in self.dtypes if i != 'O' and i != 'str']
+            if len(quantitative_columns) == 2 or len(quantitative_columns) == 3:
                 self.current_vis = VisList([i for i in self.columns], self)
             self._widget = rec_df.render_widget()
         self._recs_fresh = True