beat: import, just import sampled variables

hvasbath · Sep 29, 2021 · 4d9036c · 4d9036c
1 parent 1fa025f
commit 4d9036c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/beat/apps/beat.py b/beat/apps/beat.py
@@ -493,7 +493,10 @@ def setup(parser):
             problem = load_model(
                 options.results, options.import_from_mode,
                 hypers=False, build=False)
-            source_params = list(problem.config.problem_config.priors.keys())
+            priors = set(
+                list(problem.config.problem_config.priors.keys()))
+            rvs = set(problem.varnames)
+            source_params = list(priors.union(rvs))
 
             stage = Stage(
                 homepath=problem.outfolder,