Merge pull request #21 from JuliaStochOpt/bl/broadcastable

Defined broadcastable
JuliaStochOpt · Oct 10, 2018 · 146e135 · 146e135
2 parents febf431 + ff7a10a
commit 146e135
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/src/attributes.jl b/src/attributes.jl
@@ -24,6 +24,12 @@ attributes (properties) of transitions in the stochastic program.
 """
 abstract type AbstractTransitionAttribute end
 
+if VERSION >= v"0.7-"
+    Base.broadcastable(attr::Union{AbstractStochasticProgramAttribute,
+                                   AbstractNodeAttribute,
+                                   AbstractTransitionAttribute}) = Ref(attr)
+end
+
 """
     get(sp::AbstractStochasticProgram, attr::AbstractStochasticProgramAttribute)
 

diff --git a/src/stochprog.jl b/src/stochprog.jl
@@ -8,6 +8,9 @@ using LightGraphs
 Stochastic program instance.
 """
 abstract type AbstractStochasticProgram <: LightGraphs.AbstractGraph{Int} end
+if VERSION >= v"0.7-"
+    Base.broadcastable(sp::AbstractStochasticProgram) = Ref(sp)
+end
 
 """
     stochasticprogram(args...)