Skip to content

Commit 2363a7f

Browse files
committed
Fix types of latest operator mapping
1 parent 5c6ce35 commit 2363a7f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/OperatorEnumConstruction.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ import ..EvaluateEquationDerivativeModule: eval_grad_tree_array, _zygote_gradien
77
import ..EvaluationHelpersModule: _grad_evaluator
88

99
"""Used to set a default value for `operators` for ease of use."""
10-
@enum AvailableOperatorTypes begin
10+
@enum AvailableOperatorTypes::UInt8 begin
1111
IsNothing
1212
IsOperatorEnum
1313
IsGenericOperatorEnum
@@ -19,8 +19,8 @@ end
1919

2020
const LATEST_OPERATORS = Ref{Union{Nothing,AbstractOperatorEnum}}(nothing)
2121
const LATEST_OPERATORS_TYPE = Ref{AvailableOperatorTypes}(IsNothing)
22-
const LATEST_UNARY_OPERATOR_MAPPING = Dict{Function,Int8}()
23-
const LATEST_BINARY_OPERATOR_MAPPING = Dict{Function,Int8}()
22+
const LATEST_UNARY_OPERATOR_MAPPING = Dict{Function,fieldtype(Node{Float64}, :op)}()
23+
const LATEST_BINARY_OPERATOR_MAPPING = Dict{Function,fieldtype(Node{Float64}, :op)}()
2424
const ALREADY_DEFINED_UNARY_OPERATORS = (;
2525
operator_enum=Dict{Function,Bool}(), generic_operator_enum=Dict{Function,Bool}()
2626
)

0 commit comments

Comments
 (0)