fixed incorrect behavior in keys for Metadata

TuringLang · Nov 14, 2023 · 8b2720f · 8b2720f
1 parent 8bf6589
commit 8b2720f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/varinfo.jl b/src/varinfo.jl
@@ -878,7 +878,7 @@ _empty!(metadata) = empty!(metadata)
 end
 
 # `keys`
-Base.keys(md::Metadata) = keys(md.idcs)
+Base.keys(md::Metadata) = md.vns
 Base.keys(vi::VarInfo) = keys(vi.metadata)
 
 # HACK: Necessary to avoid returning `Any[]` which won't dispatch correctly