optional import ipython

EC-USGS · Jun 29, 2024 · 1026307 · 1026307
1 parent 1efef4a
commit 1026307
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pywatershed/plot/domain_plot.py b/pywatershed/plot/domain_plot.py
@@ -7,12 +7,12 @@
 
 import geopandas as gpd
 import pandas as pd
-from IPython.display import display
 
 from ..base import Parameters
 from ..utils import import_optional_dependency
 
 folium = import_optional_dependency("folium")
+IPython = import_optional_dependency("IPython")
 
 
 class DomainPlot:
@@ -160,7 +160,7 @@ def __init__(
     def display(self):
         if self.dom_plot is None:
             self.make_domain_plot()
-        display(self.dom_plot)
+        IPython.display(self.dom_plot)
 
     def add_hru_gdf(self, gdf):
         if self.dom_plot is not None: