fix: handle removal of Lux.NilSizePropagation.recursively_nillify

JuliaSymbolics · Nov 15, 2024 · e393216 · e393216
1 parent cf7438d
commit e393216
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ext/SymbolicsLuxExt.jl b/ext/SymbolicsLuxExt.jl
@@ -5,8 +5,10 @@ using Symbolics
 using Lux.LuxCore
 using Symbolics.SymbolicUtils
 
-function Lux.NilSizePropagation.recursively_nillify(x::SymbolicUtils.BasicSymbolic{<:Vector{<:Real}})
-    Lux.NilSizePropagation.recursively_nillify(Symbolics.wrap(x))
+@static if isdefined(Lux.NilSizePropagation, :recursively_nillify)
+    function Lux.NilSizePropagation.recursively_nillify(x::SymbolicUtils.BasicSymbolic{<:Vector{<:Real}})
+        Lux.NilSizePropagation.recursively_nillify(Symbolics.wrap(x))
+    end
 end
 
 @register_array_symbolic LuxCore.stateless_apply(