Use relative imports

flopy/discretization/unstructuredgrid.py

@@ -616,7 +616,7 @@ def plot(self, **kwargs):
         lc : matplotlib.collections.LineCollection
 
         """
-        from flopy.plot import PlotMapView
+        from ..plot import PlotMapView
 
         layer = 0
         if "layer" in kwargs:
@@ -837,7 +837,7 @@ def ncpl_from_ihc(ihc, iac):
             number of cells per plottable layer
 
         """
-        from flopy.utils.gridgen import get_ia_from_iac
+        from ..utils.gridgen import get_ia_from_iac
 
         valid = False
         ia = get_ia_from_iac(iac)

flopy/discretization/vertexgrid.py

@@ -361,7 +361,7 @@ def plot(self, **kwargs):
         lc : matplotlib.collections.LineCollection
 
         """
-        from flopy.plot import PlotMapView
+        from ..plot import PlotMapView
 
         mm = PlotMapView(modelgrid=self)
         return mm.plot_grid(**kwargs)

flopy/export/shapefile_utils.py

@@ -896,7 +896,7 @@ def get_spatialreference(epsg, text="esriwkt"):
         url : str
 
         """
-        from flopy.utils.flopy_io import get_url_text
+        from ..utils.flopy_io import get_url_text
 
         epsg_categories = (
             "epsg",

flopy/export/vtk.py

@@ -6,8 +6,8 @@
 import os
 import warnings
 import numpy as np
-from flopy.datbase import DataType, DataInterface
-from flopy.utils import Util3d
+from ..datbase import DataType, DataInterface
+from ..utils import Util3d
 
 from ..utils import import_optional_dependency
 
@@ -1836,7 +1836,7 @@ def export_heads(
     if vtk_grid_type != "auto":
         warnings.warn("vtk_grid_type is deprecated, setting to binary")
 
-    from flopy.utils import HeadFile
+    from ..utils import HeadFile
 
     if not os.path.exists(otfolder):
         os.mkdir(otfolder)
@@ -1926,7 +1926,7 @@ def export_cbc(
     if vtk_grid_type != "auto":
         warnings.warn("vtk_grid_type is deprecated, setting to binary")
 
-    from flopy.utils import CellBudgetFile
+    from ..utils import CellBudgetFile
 
     if not os.path.exists(otfolder):
         os.mkdir(otfolder)

flopy/mbase.py

@@ -16,7 +16,7 @@
 from subprocess import Popen, PIPE, STDOUT
 import copy
 import numpy as np
-from flopy import utils, discretization
+from . import utils, discretization
 from .version import __version__
 from .discretization.grid import Grid
 
@@ -1584,7 +1584,7 @@ def plot(self, SelPackList=None, **kwargs):
         >>> ml.plot()
 
         """
-        from flopy.plot import PlotUtilities
+        from .plot import PlotUtilities
 
         axes = PlotUtilities._plot_model_helper(
             self, SelPackList=SelPackList, **kwargs

flopy/mf6/data/mfdata.py

@@ -344,7 +344,7 @@ def _tas_info(tas_str):
         return None, None
 
     def export(self, f, **kwargs):
-        from flopy.export import utils
+        from ...export import utils
 
         if (
             self.data_type == DataType.array2d

flopy/mf6/data/mfdataarray.py

@@ -1403,7 +1403,7 @@ def plot(
             Empty list is returned if filename_base is not None. Otherwise
             a list of matplotlib.pyplot.axis is returned.
         """
-        from flopy.plot import PlotUtilities
+        from ...plot import PlotUtilities
 
         if not self.plottable:
             raise TypeError(
@@ -1902,7 +1902,7 @@ def plot(
             Empty list is returned if filename_base is not None. Otherwise
             a list of matplotlib.pyplot.axis is returned.
         """
-        from flopy.plot.plotutil import PlotUtilities
+        from ...plot.plotutil import PlotUtilities
 
         if not self.plottable:
             raise TypeError("Simulation level packages are not plottable")

flopy/mf6/data/mfdatalist.py

@@ -1347,7 +1347,7 @@ def plot(
             Empty list is returned if filename_base is not None. Otherwise
             a list of matplotlib.pyplot.axis is returned.
         """
-        from flopy.plot import PlotUtilities
+        from ...plot import PlotUtilities
 
         if not self.plottable:
             raise TypeError("Simulation level packages are not plottable")
@@ -1930,7 +1930,7 @@ def plot(
             Empty list is returned if filename_base is not None. Otherwise
             a list of matplotlib.pyplot.axis is returned.
         """
-        from flopy.plot import PlotUtilities
+        from ...plot import PlotUtilities
 
         if not self.plottable:
             raise TypeError("Simulation level packages are not plottable")

flopy/mf6/data/mfdatascalar.py

@@ -674,7 +674,7 @@ def plot(self, filename_base=None, file_extension=None, **kwargs):
         Returns:
              axes: list matplotlib.axes object
         """
-        from flopy.plot.plotutil import PlotUtilities
+        from ...plot.plotutil import PlotUtilities
 
         if not self.plottable:
             raise TypeError("Scalar values are not plottable")
@@ -967,7 +967,7 @@ def plot(
             Empty list is returned if filename_base is not None. Otherwise
             a list of matplotlib.pyplot.axis is returned.
         """
-        from flopy.plot.plotutil import PlotUtilities
+        from ...plot.plotutil import PlotUtilities
 
         if not self.plottable:
             raise TypeError("Simulation level packages are not plottable")

flopy/mf6/mfmodel.py

@@ -16,7 +16,7 @@
 from ..discretization.vertexgrid import VertexGrid
 from ..discretization.unstructuredgrid import UnstructuredGrid
 from ..discretization.grid import Grid
-from flopy.discretization.modeltime import ModelTime
+from ..discretization.modeltime import ModelTime
 from ..mbase import ModelInterface
 from .utils.mfenums import DiscretizationType
 from .data import mfstructure
@@ -1871,7 +1871,7 @@ def plot(self, SelPackList=None, **kwargs):
                 Empty list is returned if filename_base is not None. Otherwise
                 a list of matplotlib.pyplot.axis are returned.
         """
-        from flopy.plot.plotutil import PlotUtilities
+        from ..plot.plotutil import PlotUtilities
 
         axes = PlotUtilities._plot_model_helper(
             self, SelPackList=SelPackList, **kwargs

flopy/mf6/mfpackage.py

@@ -2868,7 +2868,7 @@ def export(self, f, **kwargs):
             None or Netcdf object
 
         """
-        from flopy import export
+        from .. import export
 
         return export.utils.package_export(f, self, **kwargs)
 
@@ -2903,7 +2903,7 @@ def plot(self, **kwargs):
             a list of matplotlib.pyplot.axis are returned.
 
         """
-        from flopy.plot.plotutil import PlotUtilities
+        from ..plot.plotutil import PlotUtilities
 
         if not self.plottable:
             raise TypeError("Simulation level packages are not plottable")

flopy/mf6/modflow/mfsimulation.py

@@ -2456,7 +2456,7 @@ def plot(self, model_list=None, SelPackList=None, **kwargs):
                 matplotlib.pyplot.axes objects
 
         """
-        from flopy.plot.plotutil import PlotUtilities
+        from ...plot.plotutil import PlotUtilities
 
         axes = PlotUtilities._plot_simulation_helper(
             self, model_list=model_list, SelPackList=SelPackList, **kwargs

flopy/mf6/utils/createpackages.py

@@ -3,6 +3,7 @@
 import datetime
 import textwrap
 from enum import Enum
+# keep below as absolute imports
 from flopy.mf6.data import mfstructure, mfdatautil
 from flopy.utils import datautil
 

flopy/modflow/mf.py

@@ -14,7 +14,7 @@
 from ..discretization.structuredgrid import StructuredGrid
 from ..discretization.unstructuredgrid import UnstructuredGrid
 from ..discretization.grid import Grid
-from flopy.discretization.modeltime import ModelTime
+from ..discretization.modeltime import ModelTime
 from .mfpar import ModflowPar
 
 

flopy/modflow/mfsfr2.py

@@ -2061,8 +2061,8 @@ def write_file(self, filename=None):
 
     def export(self, f, **kwargs):
         if isinstance(f, str) and f.lower().endswith(".shp"):
-            from flopy.utils.geometry import Polygon
-            from flopy.export.shapefile_utils import recarray2shp
+            from ..utils.geometry import Polygon
+            from ..export.shapefile_utils import recarray2shp
 
             geoms = []
             for ix, i in enumerate(self.reach_data.i):
@@ -2072,7 +2072,7 @@ def export(self, f, **kwargs):
                 geoms.append(Polygon(verts))
             recarray2shp(self.reach_data, geoms, shpname=f, **kwargs)
         else:
-            from flopy import export
+            from .. import export
 
             return export.utils.package_export(f, self, **kwargs)
 
@@ -2083,8 +2083,8 @@ def export_linkages(self, f, **kwargs):
         reaches can be used to filter for the longest connections in a GIS.
 
         """
-        from flopy.utils.geometry import LineString
-        from flopy.export.shapefile_utils import recarray2shp
+        from ..utils.geometry import LineString
+        from ..export.shapefile_utils import recarray2shp
 
         rd = self.reach_data.copy()
         m = self.parent
@@ -2126,8 +2126,8 @@ def export_outlets(self, f, **kwargs):
         the model (outset=0).
 
         """
-        from flopy.utils.geometry import Point
-        from flopy.export.shapefile_utils import recarray2shp
+        from ..utils.geometry import Point
+        from ..export.shapefile_utils import recarray2shp
 
         rd = self.reach_data
         if np.min(rd.outreach) == np.max(rd.outreach):
@@ -2160,8 +2160,8 @@ def export_transient_variable(self, f, varname, **kwargs):
             Variable in SFR Package dataset 6a (see SFR package documentation)
 
         """
-        from flopy.utils.geometry import Point
-        from flopy.export.shapefile_utils import recarray2shp
+        from ..utils.geometry import Point
+        from ..export.shapefile_utils import recarray2shp
 
         rd = self.reach_data
         if np.min(rd.outreach) == np.max(rd.outreach):

flopy/mt3d/mt.py

@@ -15,7 +15,7 @@
 from .mtsft import Mt3dSft
 from .mtlkt import Mt3dLkt
 from ..discretization.structuredgrid import StructuredGrid
-from flopy.discretization.modeltime import ModelTime
+from ..discretization.modeltime import ModelTime
 
 
 class Mt3dList(Package):

flopy/pakbase.py

@@ -676,13 +676,13 @@ def export(self, f, **kwargs):
             None or Netcdf object
 
         """
-        from flopy import export
+        from . import export
 
         return export.utils.package_export(f, self, **kwargs)
 
     def _generate_heading(self):
         """Generate heading."""
-        from flopy import __version__
+        from . import __version__
 
         parent = self.parent
         self.heading = (
@@ -821,7 +821,7 @@ def plot(self, **kwargs):
         >>> ml.dis.plot()
 
         """
-        from flopy.plot import PlotUtilities
+        from .plot import PlotUtilities
 
         if not self.plottable:
             raise TypeError(f"Package {self.name} is not plottable")

flopy/seawat/swt.py

@@ -6,7 +6,7 @@
 from .swtvdf import SeawatVdf
 from .swtvsc import SeawatVsc
 from ..discretization.structuredgrid import StructuredGrid
-from flopy.discretization.modeltime import ModelTime
+from ..discretization.modeltime import ModelTime
 
 
 class SeawatList(Package):

flopy/utils/datafile.py

@@ -382,7 +382,7 @@ def plot(
             ).transpose()
         ).transpose()
 
-        from flopy.plot.plotutil import PlotUtilities
+        from ..plot.plotutil import PlotUtilities
 
         return PlotUtilities._plot_array_helper(
             plotarray,

flopy/utils/util_array.py

@@ -653,7 +653,7 @@ def plottable(self):
         return True
 
     def export(self, f, **kwargs):
-        from flopy import export
+        from .. import export
 
         return export.utils.array3d_export(f, self, **kwargs)
 
@@ -726,7 +726,7 @@ def plot(
         >>> ml.lpf.hk.plot()
 
         """
-        from flopy.plot import PlotUtilities
+        from ..plot import PlotUtilities
 
         axes = PlotUtilities._plot_util3d_helper(
             self,
@@ -1530,7 +1530,7 @@ def plot(
         >>> ml.rch.rech.plot()
 
         """
-        from flopy.plot import PlotUtilities
+        from ..plot import PlotUtilities
 
         axes = PlotUtilities._plot_transient2d_helper(
             self,
@@ -1586,7 +1586,7 @@ def array(self):
         return arr
 
     def export(self, f, **kwargs):
-        from flopy import export
+        from .. import export
 
         return export.utils.transient2d_export(f, self, **kwargs)
 
@@ -1984,7 +1984,7 @@ def plot(
         >>> ml.dis.top.plot()
 
         """
-        from flopy.plot import PlotUtilities
+        from ..plot import PlotUtilities
 
         axes = PlotUtilities._plot_util2d_helper(
             self,
@@ -1997,7 +1997,7 @@ def plot(
         return axes
 
     def export(self, f, **kwargs):
-        from flopy import export
+        from .. import export
 
         return export.utils.array2d_export(f, self, **kwargs)
 

flopy/utils/util_list.py

@@ -123,7 +123,7 @@ def get_empty(self, ncell=0):
         return d
 
     def export(self, f, **kwargs):
-        from flopy import export
+        from .. import export
 
         return export.utils.mflist_export(f, self, **kwargs)
 
@@ -943,7 +943,7 @@ def plot(
 
         """
 
-        from flopy.plot import PlotUtilities
+        from ..plot import PlotUtilities
 
         axes = PlotUtilities._plot_mflist_helper(
             self,