import and use Setifled's @set

Author: TorkelE

Project.toml

@@ -17,6 +17,7 @@ Parameters = "d96e819e-fc66-5662-9728-84c9c7592b0a"
 Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
 Requires = "ae029012-a4dd-5104-9daa-d747884805df"
 RuntimeGeneratedFunctions = "7e49a35a-f44a-4d26-94aa-eba1b4ca6b47"
+Setfield = "efcf1570-3423-57d1-acb7-fd33fddbac46"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 SymbolicUtils = "d1185830-fcd6-423d-90d6-eec64667417b"
 Symbolics = "0c5d862f-8b57-4792-8d23-62f2024744c7"

ext/CatalystHomotopyContinuationExtension.jl

@@ -4,6 +4,7 @@ module CatalystHomotopyContinuationExtension
 using Catalyst
 import ModelingToolkit as MT
 import HomotopyContinuation as HC
+import Setfield: @set
 import Symbolics: unwrap, wrap, Rewriters, symtype, issym, istree
 
 # Creates and exports hc_steady_states function.

ext/CatalystHomotopyContinuationExtension/homotopy_continuation_extension.jl

@@ -45,7 +45,7 @@ end
 # For a given reaction system, parameter values, and initial conditions, find the polynomial that HC solves to find steady states.
 function steady_state_polynomial(rs_in::ReactionSystem, ps, u0)
     any(!(eq isa Reaction) for eq in rs_in.eqs) && error("This feature is currently not supported for reaction systems containing (non-reaction) equations.")
-    rs = ModelingToolkit.@set rs_in.rxs = [Catalyst.expand_registered_functions(rx) for rx in rs_in.rxs]
+    rs = @set rs_in.rxs = [Catalyst.expand_registered_functions(rx) for rx in rs_in.rxs]
     ns = convert(NonlinearSystem, rs; remove_conserved = true)
     pre_varmap = [symmap_to_varmap(rs,u0)..., symmap_to_varmap(rs,ps)...]
     conservationlaw_errorcheck(rs, pre_varmap)

src/Catalyst.jl

@@ -4,7 +4,7 @@ $(DocStringExtensions.README)
 module Catalyst
 
 using DocStringExtensions
-using SparseArrays, DiffEqBase, Reexport
+using SparseArrays, DiffEqBase, Reexport, Setfield
 using LaTeXStrings, Latexify, Requires
 using JumpProcesses: JumpProcesses,
                      JumpProblem, MassActionJump, ConstantRateJump,

test/extensions/homotopy_continuation.jl

@@ -93,7 +93,7 @@ let
     f = ODEFunction(convert(ODESystem, rs))
     @test length(sss) == 4
     for ss in sss
-        @test f(sss[1], last.(ps), 0.0)[1] == 0.0
+        @test isapprox(f(sss[1], last.(ps), 0.0)[1], 0.0; atol=1e-12)
     end
 
     @test_throws Exception hc_steady_states(rs, [:v => 5.0, :K => 2.5, :n => 2.7, :d => 1.0]; show_progress=false)