update

SciML · Oct 10, 2024 · d3ae594 · d3ae594
1 parent 526fd0c
commit d3ae594
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/NeuralPDE.jl b/src/NeuralPDE.jl
@@ -34,6 +34,7 @@ import ChainRulesCore, Lux, ComponentArrays
 using Lux: FromFluxAdaptor, recursive_eltype
 using ChainRulesCore: @non_differentiable
 using NeuralOperators
+import Lux: Chain
 import NeuralOperators: DeepONet
 RuntimeGeneratedFunctions.init(@__MODULE__)