Merge pull request #8227 from OpenMined/fix-asset-mock-repr

fix check for mock if its a DataFrame in _repr_html_
OpenMined · Nov 9, 2023 · 7885703 · 7885703
2 parents 7b20f59 + 8f514c4
commit 7885703
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/syft/src/syft/service/dataset/dataset.py b/packages/syft/src/syft/service/dataset/dataset.py
@@ -176,7 +176,7 @@ def _repr_html_(self) -> Any:
             mock_table_line = itables.to_html_datatable(
                 df=self.mock.syft_action_data, css=itables_css
             )
-        elif isinstance(self.data, pd.DataFrame):
+        elif isinstance(self.mock, pd.DataFrame):
             mock_table_line = itables.to_html_datatable(df=self.mock, css=itables_css)
         else:
             mock_table_line = self.mock