Skip to content

Commit

Permalink
absolute import of strat in the plot module
Browse files Browse the repository at this point in the history
  • Loading branch information
mcflugen committed Dec 4, 2024
1 parent e7cf233 commit d3636fa
Showing 1 changed file with 10 additions and 6 deletions.
16 changes: 10 additions & 6 deletions deltametrics/plot.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,11 @@
import matplotlib.collections as coll
import mpl_toolkits.axes_grid1 as axtk

from . import strat
from deltametrics.strat import _adjust_elevation_by_subsidence
from deltametrics.strat import _compute_elevation_to_preservation
from deltametrics.strat import _compute_preservation_to_cube
from deltametrics.strat import _compute_preservation_to_time_intervals
from deltametrics.strat import _determine_strat_coordinates

# from . import section

Expand Down Expand Up @@ -1181,11 +1185,11 @@ def show_one_dimensional_trajectory_to_strata(

if sigma_dist is not None:
# adjust elevations by subsidence rate
e = strat._adjust_elevation_by_subsidence(e_in, sigma_dist)
s, p = strat._compute_elevation_to_preservation(e) # strat, preservation
z = strat._determine_strat_coordinates(e, dz=dz, z=z, nz=nz) # vert coordinates
e = _adjust_elevation_by_subsidence(e_in, sigma_dist)
s, p = _compute_elevation_to_preservation(e) # strat, preservation
z = _determine_strat_coordinates(e, dz=dz, z=z, nz=nz) # vert coordinates
z_disp = np.append(z, z[-1])
sc, dc = strat._compute_preservation_to_cube(s, z)
sc, dc = _compute_preservation_to_cube(s, z)
lst = np.argmin(s < s[-1]) # last elevation

c = np.full_like(z, np.nan)
Expand All @@ -1201,7 +1205,7 @@ def show_one_dimensional_trajectory_to_strata(
pt[
np.union1d(
p.nonzero()[0],
np.array(strat._compute_preservation_to_time_intervals(p).nonzero()[0]),
np.array(_compute_preservation_to_time_intervals(p).nonzero()[0]),
)
] = 1
ax.add_collection(
Expand Down

0 comments on commit d3636fa

Please sign in to comment.