Support flattening vectors of prov.

PumasAI · Dec 8, 2020 · b66ffda · b66ffda
1 parent 94c7331
commit b66ffda
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/ProvenanceBase.jl b/src/ProvenanceBase.jl
@@ -182,6 +182,12 @@ function flat!(out, nt::NamedTuple, root = ())
     return out
 end
 flat!(out, ::Nothing, root = ()) = out
+function flat!(out, v::AbstractVector, root)
+    for (nth, each) in enumerate(v)
+        flat!(out, each, (root..., nth))
+    end
+    return out
+end
 flat!(out, value, root) = (out[root] = value; out)
 flat!(out, p::Provenance, root) = flat!(out, p.data, root)