minor fix

SciML · May 17, 2024 · 256a6f3 · 256a6f3
1 parent a1ce8e8
commit 256a6f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/NeuralPDE.jl b/src/NeuralPDE.jl
@@ -31,7 +31,7 @@ using DomainSets: Domain, ClosedInterval, AbstractInterval, leftendpoint, righte
 using SciMLBase: @add_kwonly, parameterless_type
 using UnPack: @unpack
 import ChainRulesCore, Lux, ComponentArrays
-# using Lux: FromFluxAdaptor
+using Lux: FromFluxAdaptor
 using ChainRulesCore: @non_differentiable
 
 RuntimeGeneratedFunctions.init(@__MODULE__)