Fix broadcasting of inputs (#238)

FriesischScott · Dec 5, 2024 · fa6c842 · fa6c842
1 parent c9d89c2
commit fa6c842
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/inputs/inputs.jl b/src/inputs/inputs.jl
@@ -56,3 +56,5 @@ function sns_zero_point(inputs::AbstractVector{<:UQInput})
 
     return sns
 end
+
+Base.broadcastable(i::T) where {T<:UQInput} = Ref(i)
diff --git a/test/inputs/inputs.jl b/test/inputs/inputs.jl
@@ -31,4 +31,10 @@ inputs = [pi, jd, z]
     @testset "count_rvs" begin
         @test count_rvs(inputs) == 3
     end
+
+    @testset "broadcasting" begin
+        x = RandomVariable(Normal(), :x)
+        u = rand(10)
+        @test pdf.(x.dist, u) == pdf.(x, u)
+    end
 end
Original file line number	Diff line number	Diff line change
Expand Up		@@ -56,3 +56,5 @@ function sns_zero_point(inputs::AbstractVector{<:UQInput})

		return sns
		end

		Base.broadcastable(i::T) where {T<:UQInput} = Ref(i)