diff --git a/pynecone/components/datadisplay/datatable.py b/pynecone/components/datadisplay/datatable.py index e7b4f6bd0..bed81c1ac 100644 --- a/pynecone/components/datadisplay/datatable.py +++ b/pynecone/components/datadisplay/datatable.py @@ -69,11 +69,6 @@ class DataTable(Gridjs): ) def _render(self) -> Tag: - # If given a pandas df break up the data and columns - if utils.is_dataframe(type(self.data)): - self.columns = Var.create(list(self.data.columns.values.tolist())) # type: ignore - self.data = Var.create(list(self.data.values.tolist())) # type: ignore - # If given a var dataframe, get the data and columns if isinstance(self.data, Var): self.columns = BaseVar( @@ -87,5 +82,10 @@ class DataTable(Gridjs): state=self.data.state, ) + # If given a pandas df break up the data and columns + if utils.is_dataframe(type(self.data)): + self.columns = Var.create(list(self.data.columns.values.tolist())) # type: ignore + self.data = Var.create(list(self.data.values.tolist())) # type: ignore + # Render the table. return super()._render()