ADSDataset.show_in_notebook fails while rendering visualizations (#777)

VipulMascarenhas · web-flow · commit 58cb61f7e5cc · 2024-04-17T09:32:19.000-07:00
diff --git a/ads/catalog/model.py b/ads/catalog/model.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 # -*- coding: utf-8; -*-
 
-# Copyright (c) 2020, 2023 Oracle and/or its affiliates.
+# Copyright (c) 2020, 2024 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 import warnings
@@ -404,13 +404,13 @@ def show_in_notebook(self, display_format: str = "dataframe") -> None:
     def _repr_html_(self):
         """Shows model in dataframe format."""
         return (
-            self.to_dataframe().style.set_properties(**{"margin-left": "0px"}).render()
+            self.to_dataframe().style.set_properties(**{"margin-left": "0px"}).to_html()
         )
 
     def __repr__(self):
         """Shows model in dataframe format."""
         return (
-            self.to_dataframe().style.set_properties(**{"margin-left": "0px"}).render()
+            self.to_dataframe().style.set_properties(**{"margin-left": "0px"}).to_html()
         )
 
     def activate(self) -> None:
diff --git a/ads/catalog/notebook.py b/ads/catalog/notebook.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 # -*- coding: utf-8; -*-
 
-# Copyright (c) 2020, 2022 Oracle and/or its affiliates.
+# Copyright (c) 2020, 2024 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 import warnings
@@ -220,7 +220,7 @@ def _repr_html_(notebook_self):
                 lambda x: "<a href='%s'>%s</a>"
                 % (x if x.startswith("http") else "http://%s" % x, "open")
             )
-            return df.style.set_properties(**{"margin-left": "0px"}).render()
+            return df.style.set_properties(**{"margin-left": "0px"}).to_html()
 
         notebook.commit = MethodType(commit, notebook)
         notebook.rollback = MethodType(rollback, notebook)
@@ -295,7 +295,7 @@ def create_notebook_session(
         shape=None,
         block_storage_size_in_gbs=None,
         subnet_id=None,
-        **kwargs
+        **kwargs,
     ):
         """
         Create a new notebook session with the supplied details.
diff --git a/ads/catalog/project.py b/ads/catalog/project.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 # -*- coding: utf-8; -*-
 
-# Copyright (c) 2020, 2022 Oracle and/or its affiliates.
+# Copyright (c) 2020, 2024 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 import warnings
@@ -237,7 +237,7 @@ def _repr_html_(project_self):
             return (
                 project_self.to_dataframe()
                 .style.set_properties(**{"margin-left": "0px"})
-                .render()
+                .to_html()
             )
 
         project.commit = MethodType(commit, project)
diff --git a/ads/catalog/summary.py b/ads/catalog/summary.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*--
 
-# Copyright (c) 2020, 2022 Oracle and/or its affiliates.
+# Copyright (c) 2020, 2024 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 from __future__ import print_function, absolute_import
@@ -98,7 +98,6 @@ def __add__(self, rhs):  # pragma: no cover
         pass
 
     def to_dataframe(self, datetime_format=None):
-
         """
         Returns the model catalog summary as a pandas dataframe
 
@@ -121,7 +120,6 @@ def to_dataframe(self, datetime_format=None):
 
     @runtime_dependency(module="IPython", install_from=OptionalDependency.NOTEBOOK)
     def show_in_notebook(self, datetime_format=None):
-
         """
         Displays the model catalog summary in a Jupyter Notebook cell
 
@@ -144,7 +142,7 @@ def show_in_notebook(self, datetime_format=None):
     def _repr_html_(self):
         return self.df.style.applymap(
             self._color_lifecycle_state, subset=["lifecycle_state"]
-        ).render()
+        ).to_html()
 
     def _sort_by(self, cols, reverse=False):
         return sorted(
diff --git a/ads/data_labeling/metadata.py b/ads/data_labeling/metadata.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 # -*- coding: utf-8; -*-
 
-# Copyright (c) 2021, 2022 Oracle and/or its affiliates.
+# Copyright (c) 2021, 2024 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 from dataclasses import asdict, dataclass, field
@@ -75,7 +75,7 @@ def to_dataframe(self) -> pd.DataFrame:
     def _repr_html_(self):
         """Shows metadata in dataframe format."""
         return (
-            self.to_dataframe().style.set_properties(**{"margin-left": "0px"}).render()
+            self.to_dataframe().style.set_properties(**{"margin-left": "0px"}).to_html()
         )
 
     @classmethod
diff --git a/ads/dataset/dataset.py b/ads/dataset/dataset.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*--
 
-# Copyright (c) 2020, 2023 Oracle and/or its affiliates.
+# Copyright (c) 2020, 2024 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 from __future__ import print_function, absolute_import, division
@@ -85,7 +85,6 @@ def __init__(
         interactive=False,
         **kwargs,
     ):
-
         #
         # to keep performance high and linear no matter the size of the distributed dataset we
         # create a pandas df that's used internally because this has a fixed upper size.
@@ -204,7 +203,7 @@ def _repr_html_(self):
                     .style.set_table_styles(utils.get_dataframe_styles())
                     .set_table_attributes("class=table")
                     .hide_index()
-                    .render()
+                    .to_html()
                 )
             )
         )
@@ -263,7 +262,7 @@ def _repr_html_(self):
                             self.style.set_table_styles(utils.get_dataframe_styles())
                             .set_table_attributes("class=table")
                             .hide_index()
-                            .render()
+                            .to_html()
                         )
                     )
                 )
@@ -1265,7 +1264,6 @@ def _build_new_dataset(
         n=None,
         **init_kwargs,
     ):
-
         prev_doc_mode = utils.is_documentation_mode()
 
         set_documentation_mode(False)
diff --git a/ads/dataset/factory.py b/ads/dataset/factory.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 # -*- coding: utf-8; -*-
 
-# Copyright (c) 2020, 2022 Oracle and/or its affiliates.
+# Copyright (c) 2020, 2024 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 from __future__ import print_function, absolute_import
@@ -367,7 +367,7 @@ def list_snapshots(snapshot_dir=None, name="", storage_options=None, **kwargs):
                 HTML(
                     list_df.style.set_table_attributes("class=table")
                     .hide_index()
-                    .render()
+                    .to_html()
                 )
             )
         return list_df
@@ -884,7 +884,6 @@ def read_xml(path: str, **kwargs) -> pd.DataFrame:
         import xml.etree.cElementTree as et
 
         def get_children(df, node, parent, i):
-
             for name in node.attrib.keys():
                 df.at[i, parent + name] = node.attrib[name]
             for child in list(node):
diff --git a/ads/dataset/sampled_dataset.py b/ads/dataset/sampled_dataset.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 # -*- coding: utf-8; -*-
 
-# Copyright (c) 2020, 2022 Oracle and/or its affiliates.
+# Copyright (c) 2020, 2024 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 import matplotlib
@@ -49,6 +49,7 @@
 
 NATURAL_EARTH_DATASET = "naturalearth_lowres"
 
+
 class PandasDataset(object):
     """
     This class provides APIs that can work on a sampled dataset.
@@ -107,7 +108,6 @@ def __init__(
         self.sampled_df = self.sampled_df.reset_index(drop=True)
 
     def _find_feature_subset(self, df, target_name, include_n_features=32):
-
         if len(df.columns) <= include_n_features:
             return self.sampled_df
         else:
@@ -212,7 +212,6 @@ def _calculate_dataset_statistics(self, is_wide_dataset, out):
     def _generate_features_html(
         self, is_wide_dataset, n_features, df_stats, visualizations_follow
     ):
-
         html = utils.get_bootstrap_styles()
 
         if is_wide_dataset:
@@ -233,7 +232,7 @@ def _generate_features_html(
                 if ("float" in str(type(x))) or ("int" in str(type(x)))
                 else x
             )
-            .render()
+            .to_html()
         )
 
         if visualizations_follow:
@@ -244,7 +243,6 @@ def _generate_features_html(
     def _generate_warnings_html(
         self, is_wide_dataset, n_rows, n_features, df_stats, out, accordion
     ):
-
         #
         # create the "Warnings" accordion section:
         #  - show high cardinal categoricals

Original file line number	Diff line number	Diff line change
`@@ -1,7 +1,7 @@`
`1`	`1`	`#!/usr/bin/env python`
`2`	`2`	`# -- coding: utf-8 ---`
`3`	`3`
`4`		`-# Copyright (c) 2020, 2023 Oracle and/or its affiliates.`
	`4`	`+# Copyright (c) 2020, 2024 Oracle and/or its affiliates.`
`5`	`5`	`# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/`
`6`	`6`
`7`	`7`	`from __future__ import print_function, absolute_import, division`
`@@ -85,7 +85,6 @@ def __init__(`
`85`	`85`	`interactive=False,`
`86`	`86`	`**kwargs,`
`87`	`87`	`):`
`88`		`-`
`89`	`88`	`#`
`90`	`89`	`# to keep performance high and linear no matter the size of the distributed dataset we`
`91`	`90`	`# create a pandas df that's used internally because this has a fixed upper size.`
`@@ -204,7 +203,7 @@ def _repr_html_(self):`
`204`	`203`	`.style.set_table_styles(utils.get_dataframe_styles())`
`205`	`204`	`.set_table_attributes("class=table")`
`206`	`205`	`.hide_index()`
`207`		`- .render()`
	`206`	`+ .to_html()`
`208`	`207`	`)`
`209`	`208`	`)`
`210`	`209`	`)`
`@@ -263,7 +262,7 @@ def _repr_html_(self):`
`263`	`262`	`self.style.set_table_styles(utils.get_dataframe_styles())`
`264`	`263`	`.set_table_attributes("class=table")`
`265`	`264`	`.hide_index()`
`266`		`- .render()`
	`265`	`+ .to_html()`
`267`	`266`	`)`
`268`	`267`	`)`
`269`	`268`	`)`
`@@ -1265,7 +1264,6 @@ def _build_new_dataset(`
`1265`	`1264`	`n=None,`
`1266`	`1265`	`**init_kwargs,`
`1267`	`1266`	`):`
`1268`		`-`
`1269`	`1267`	`prev_doc_mode = utils.is_documentation_mode()`
`1270`	`1268`
`1271`	`1269`	`set_documentation_mode(False)`