Skip to content

Commit c6623c4

Browse files
committed
fix: get_state/set_state
1 parent ca23047 commit c6623c4

File tree

2 files changed

+11
-4
lines changed

2 files changed

+11
-4
lines changed

src/GraphDynamicalSystems.jl

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,8 @@ export QualitativeNetwork,
2525
target_functions,
2626
interpret,
2727
create_qn_system,
28-
default_target_function
28+
default_target_function,
29+
set_state!,
30+
current_parameters
2931

3032
end

src/qualitative_networks.jl

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,11 @@
1-
import DynamicalSystemsBase: get_state, set_state!
1+
import DynamicalSystemsBase: get_state, set_state!, current_parameters
22
import JSON
33
import SciMLBase
44
import StructUtils
55

66
using AbstractTrees: Leaves, PostOrderDFS
77
using AutoHashEquals: @auto_hash_equals
8-
using DynamicalSystemsBase: ArbitrarySteppable, current_parameters, initial_state
8+
using DynamicalSystemsBase: ArbitrarySteppable, initial_state
99
using Graphs: AbstractGraph, SimpleDiGraph, add_edge!, add_vertex!, ne
1010
using HerbConstraints: DomainRuleNode, Forbidden, Ordered, Unique, VarNode, addconstraint!
1111
using HerbCore: AbstractGrammar, RuleNode, get_rule
@@ -555,6 +555,10 @@ function set_state!(qn::QN, entity::Symbol, value::Integer)
555555
set_state!(qn, EntityName(entity), value)
556556
end
557557

558+
function set_state!(qn::QN, values)
559+
set_state!.((qn,), entities(qn), values)
560+
end
561+
558562
"""
559563
$(TYPEDSIGNATURES)
560564
@@ -630,6 +634,7 @@ end
630634

631635
extract_state(model::QN) = model.state
632636
extract_parameters(model::QN) = model.graph
637+
current_parameters(model::QN) = model.graph
633638
reset_model!(model::QN, u, _) = model.state .= u
634639

635640
function SciMLBase.reinit!(
@@ -658,7 +663,7 @@ function create_qn_system(qn::QN)
658663
extract_state,
659664
extract_parameters,
660665
reset_model!,
661-
isdeterministic = false,
666+
isdeterministic = get_schedule(qn) == Synchronous(),
662667
)
663668
end
664669

0 commit comments

Comments
 (0)