Moved geopandas file load out of constructor

oracle · Sep 19, 2023 · 6b52253 · 6b52253
1 parent 4b55395
commit 6b52253
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/ads/dataset/sampled_dataset.py b/ads/dataset/sampled_dataset.py
@@ -67,9 +67,6 @@ def __init__(
         self.correlation = None
         self.feature_dist_html_dict = {}
         self.feature_types = metadata if metadata is not None else {}
-        self.world = geopandas.read_file(
-            geopandas.datasets.get_path("naturalearth_lowres")
-        )
 
         self.numeric_columns = self.sampled_df.select_dtypes(
             utils.numeric_pandas_dtypes()
@@ -706,6 +703,11 @@ def _visualize_feature_distribution(self, html_widget):
                 gdf = geopandas.GeoDataFrame(
                     df, geometry=geopandas.points_from_xy(df["lon"], df["lat"])
                 )
+
+                self.world = geopandas.read_file(
+                    geopandas.datasets.get_path("naturalearth_lowres")
+                )
+
                 self.world.plot(
                     ax=ax, color="lightgrey", linewidth=0.5, edgecolor="white"
                 )