ParTable: graph-based ctor supports params= kw

Author: alyst

src/frontend/specification/StenoGraphs.jl

@@ -32,8 +32,9 @@ label(args...) = Label(args)
 
 function ParameterTable(
     graph::AbstractStenoGraph;
-    observed_vars,
-    latent_vars,
+    observed_vars::AbstractVector{Symbol},
+    latent_vars::AbstractVector{Symbol},
+    params::Union{AbstractVector{Symbol}, Nothing} = nothing,
     group::Integer = 1,
     param_prefix = :θ,
 )
@@ -103,7 +104,7 @@ function ParameterTable(
         end
     end
 
-    return ParameterTable(columns; latent_vars, observed_vars)
+    return ParameterTable(columns; latent_vars, observed_vars, params)
 end
 
 ############################################################################################
@@ -112,20 +113,23 @@ end
 
 function EnsembleParameterTable(
     graph::AbstractStenoGraph;
-    observed_vars,
-    latent_vars,
+    observed_vars::AbstractVector{Symbol},
+    latent_vars::AbstractVector{Symbol},
+    params::Union{AbstractVector{Symbol}, Nothing} = nothing,
     groups,
 )
     graph = unique(graph)
 
     partables = Dict(
         group => ParameterTable(
             graph;
-            observed_vars = observed_vars,
-            latent_vars = latent_vars,
+            observed_vars,
+            latent_vars,
+            params,
             group = i,
             param_prefix = Symbol(:g, group),
         ) for (i, group) in enumerate(groups)
     )
-    return EnsembleParameterTable(partables)
+
+    return EnsembleParameterTable(partables; params)
 end