Comments adressed

core/Project.toml

@@ -74,6 +74,7 @@ Legolas = "0.5"
 LinearSolve = "2.24"
 Logging = "<0.0.1, 1"
 LoggingExtras = "1"
+Makie = "0.21"
 MetaGraphsNext = "0.6, 0.7"
 OrdinaryDiffEq = "6.7"
 PreallocationTools = "0.4"

core/ext/RibasimMakieExt.jl

@@ -1,12 +1,7 @@
 module RibasimMakieExt
+using DataFrames: DataFrame
+using Makie: Figure, Axis, lines!, axislegend
 using Ribasim: Ribasim, Model
-if isdefined(Base, :get_extension)
-    using Makie
-    using DataFrames
-else
-    using ..Makie
-    using ..DataFrames
-end
 
 function Ribasim.plot_basin_data!(model::Model, ax::Axis, column::Symbol)
     basin_data = DataFrame(Ribasim.basin_table(model))
@@ -21,8 +16,8 @@ end
 
 function Ribasim.plot_basin_data(model::Model)
     f = Figure()
-    ax1 = Axis(f[1, 1]; ylabel = "level")
-    ax2 = Axis(f[2, 1]; xlabel = "time", ylabel = "storage")
+    ax1 = Axis(f[1, 1]; ylabel = "level [m]")
+    ax2 = Axis(f[2, 1]; xlabel = "time", ylabel = "storage [m³]")
     Ribasim.plot_basin_data!(model, ax1, :level)
     Ribasim.plot_basin_data!(model, ax2, :storage)
     f
@@ -49,7 +44,7 @@ end
 
 function Ribasim.plot_flow(model::Model)
     f = Figure()
-    ax = Axis(f[1, 1]; xlabel = "time", ylabel = "flow rate")
+    ax = Axis(f[1, 1]; xlabel = "time", ylabel = "flow rate [m³s⁻¹]")
     edge_ids = unique(Ribasim.flow_table(model).edge_id)
     for edge_id in edge_ids
         Ribasim.plot_flow!(model, ax, edge_id; skip_conservative_out = true)