fix shallow water example

CliMA · Mar 8, 2025 · 1022e6a · 1022e6a
1 parent 9df71e1
commit 1022e6a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/examples/shallow_water_Bickley_jet.jl b/examples/shallow_water_Bickley_jet.jl
@@ -178,11 +178,11 @@ ds = NCDataset(simulation.output_writers[:fields].filepath, "r")
 
 times = ds["time"][:]
 
-ω = @lift ds["ω"][:, :, 1, $n]
+ω = @lift ds["ω"][:, :, $n]
 hm_ω = heatmap!(ax_ω, x, y, ω, colorrange = (-1, 1), colormap = :balance)
 Colorbar(fig[2, 2], hm_ω)
 
-ω′ = @lift ds["ω′"][:, :, 1, $n]
+ω′ = @lift ds["ω′"][:, :, $n]
 hm_ω′ = heatmap!(ax_ω′, x, y, ω′, colormap = :balance)
 Colorbar(fig[2, 4], hm_ω′)
 

diff --git a/src/OutputWriters/netcdf_output_writer.jl b/src/OutputWriters/netcdf_output_writer.jl
@@ -7,6 +7,7 @@ using Oceananigans.Fields
 using Oceananigans.Grids: AbstractCurvilinearGrid, RectilinearGrid, StaticVerticalDiscretization
 using Oceananigans.Grids: topology, halo_size, parent_index_range, ξnodes, ηnodes, rnodes, validate_index, peripheral_node
 using Oceananigans.Fields: reduced_dimensions, reduced_location, location
+using Oceananigans.Models: ShallowWaterModel
 using Oceananigans.ImmersedBoundaries: ImmersedBoundaryGrid, GridFittedBottom, GFBIBG, GridFittedBoundary
 using Oceananigans.TimeSteppers: float_or_date_time
 using Oceananigans.BuoyancyFormulations: BuoyancyForce, BuoyancyTracer, SeawaterBuoyancy, LinearEquationOfState
@@ -605,7 +606,8 @@ default_tracer_attributes(::SeawaterBuoyancy) = Dict(
 
 function default_output_attributes(model)
     velocity_attrs = default_velocity_attributes(model.grid)
-    tracer_attrs = default_tracer_attributes(model.buoyancy)
+    buoyancy = model isa ShallowWaterModel ? nothing : model.buoyancy
+    tracer_attrs = default_tracer_attributes(buoyancy)
     return merge(velocity_attrs, tracer_attrs)
 end