Fix varinfo[:] for empty varinfo (#926)

* Fix varinfo[:] for empty varinfo

* Update changelog, bump patch

* Function no longer mutates, remove !!

* No need to empty!! if it's not mutated

* `vi` escapes the scope of the function...

Author: penelopeysm

HISTORY.md

@@ -1,5 +1,9 @@
 # DynamicPPL Changelog
 
+## 0.36.5
+
+`varinfo[:]` now returns an empty vector if `varinfo::DynamicPPL.NTVarInfo` is empty, rather than erroring.
+
 ## 0.36.4
 
 Added compatibility with DifferentiationInterface.jl 0.7, and also with JET.jl 0.10.

Project.toml

@@ -1,6 +1,6 @@
 name = "DynamicPPL"
 uuid = "366bfd00-2699-11ea-058f-f148b4cae6d8"
-version = "0.36.4"
+version = "0.36.5"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"

src/varinfo.jl

@@ -854,6 +854,9 @@ getindex_internal(vi::VarInfo, ::Colon) = getindex_internal(vi.metadata, Colon()
 function getindex_internal(vi::NTVarInfo, ::Colon)
     return reduce(vcat, map(Base.Fix2(getindex_internal, Colon()), vi.metadata))
 end
+function getindex_internal(vi::VarInfo{NamedTuple{(),Tuple{}}}, ::Colon)
+    return float(Real)[]
+end
 function getindex_internal(md::Metadata, ::Colon)
     return mapreduce(
         Base.Fix1(getindex_internal, md), vcat, md.vns; init=similar(md.vals, 0)

test/varinfo.jl

@@ -78,8 +78,8 @@ end
         @test vn2 == vn1
         @test hash(vn2) == hash(vn1)
 
-        function test_base!!(vi_original)
-            vi = empty!!(vi_original)
+        function test_base(vi_original)
+            vi = deepcopy(vi_original)
             @test getlogp(vi) == 0
             @test isempty(vi[:])
 
@@ -97,8 +97,10 @@ end
 
             @test length(vi[vn]) == 1
             @test vi[vn] == r
+            @test vi[:] == [r]
             vi = DynamicPPL.setindex!!(vi, 2 * r, vn)
             @test vi[vn] == 2 * r
+            @test vi[:] == [2 * r]
 
             # TODO(mhauru) Implement these functions for other VarInfo types too.
             if vi isa DynamicPPL.UntypedVectorVarInfo
@@ -113,12 +115,11 @@ end
             @test ~isempty(vi)
         end
 
-        vi = VarInfo()
-        test_base!!(vi)
-        test_base!!(DynamicPPL.typed_varinfo(vi))
-        test_base!!(SimpleVarInfo())
-        test_base!!(SimpleVarInfo(Dict()))
-        test_base!!(SimpleVarInfo(DynamicPPL.VarNamedVector()))
+        test_base(VarInfo())
+        test_base(DynamicPPL.typed_varinfo(VarInfo()))
+        test_base(SimpleVarInfo())
+        test_base(SimpleVarInfo(Dict()))
+        test_base(SimpleVarInfo(DynamicPPL.VarNamedVector()))
     end
 
     @testset "get/set/acc/resetlogp" begin