diff --git a/echopype/visualize/__init__.py b/echopype/visualize/__init__.py index 7144aa75d..556917709 100644 --- a/echopype/visualize/__init__.py +++ b/echopype/visualize/__init__.py @@ -1,4 +1,4 @@ -from echopype.visualize.api import create_echogram -from echopype.visualize import cm +from .api import create_echogram +from . import cm __all__ = ["create_echogram", "cm"] diff --git a/echopype/visualize/api.py b/echopype/visualize/api.py index 2fbedbbe6..2369049a6 100644 --- a/echopype/visualize/api.py +++ b/echopype/visualize/api.py @@ -3,8 +3,8 @@ import xarray as xr -from echopype.visualize.plot import _plot_echogram, FacetGrid, QuadMesh, T -from echopype.echodata import EchoData +from .plot import _plot_echogram, FacetGrid, QuadMesh, T +from ..echodata import EchoData def create_echogram( diff --git a/echopype/visualize/plot.py b/echopype/visualize/plot.py index 6e23af5a2..6626e3cf3 100644 --- a/echopype/visualize/plot.py +++ b/echopype/visualize/plot.py @@ -5,8 +5,8 @@ import numpy as np from xarray.plot.facetgrid import FacetGrid from matplotlib.collections import QuadMesh -from echopype.visualize.cm import cmap_d from typing import Optional, Union, List, TypeVar +from .cm import cmap_d T = TypeVar('T', int, float)