correected adapts again

jagoosw · Nov 28, 2023 · 355d9bd · 355d9bd
1 parent 68c1cc8
commit 355d9bd
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 3 deletions.
diff --git a/src/Walrus.jl b/src/Walrus.jl
@@ -8,8 +8,6 @@ export Tide, TidalForcing
 
 using Adapt
 
-import Adapt: adapt_structure
-
 include("wall_model.jl")
 include("radiative_transfer/radiative_transfer.jl")
 include("tidal_forcings.jl")

diff --git a/src/radiative_transfer/radiative_transfer.jl b/src/radiative_transfer/radiative_transfer.jl
@@ -19,6 +19,7 @@ using KernelAbstractions.Extras: @unroll
 
 import Oceananigans.Biogeochemistry: update_biogeochemical_state!, update_tendencies!, AbstractBiogeochemistry
 
+import Adapt: adapt_structure
 import Base: summary, show
 
 RadiationField(grid, surface_flux) = CenterField(grid; 

diff --git a/src/tidal_forcings.jl b/src/tidal_forcings.jl
@@ -11,6 +11,8 @@ using Oceananigans.Coriolis: fᶠᶠᵃ, AbstractRotation
 using Oceananigans.Forcings: Forcing
 using Oceananigans.Units: hours
 
+import Adapt: adapt_structure
+
 """
     Tide(; x_amplitude, 
            y_amplitude, 

diff --git a/src/wall_model.jl b/src/wall_model.jl
@@ -15,6 +15,7 @@ using Oceananigans.BoundaryConditions: FluxBoundaryCondition
 using Oceananigans.Fields: Center, Face
 using Oceananigans.Grids: znode
 
+import Adapt: adapt_structure
 import Base: summary, show
 
 """
@@ -70,7 +71,6 @@ julia> boundary_conditions = (u = FieldBoundaryConditions(bottom = FluxBoundaryC
                       B :: FT = 5.2
 end
 
-adapt(to, ws::WallStress) = ws
 adapt_structure(to, ws::WallStress) = ws
 
 @inline stress_velocity(uₜ, params) = log(params.z₁ * uₜ / (params.ν + eps(0.0))) / (params.κ + eps(0.0)) + params.B - params.U₁ / (uₜ + eps(0.0))