return state instead of paramStore

pinellolab · Mar 30, 2024 · d29be47 · d29be47
1 parent c7beb55
commit d29be47
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/bean/cli/run.py b/bean/cli/run.py
@@ -167,8 +167,10 @@ def main(args):
                 f"Using {len(negctrl_idx)} negative control elements to adjust phenotypic effect sizes..."
             )
             ndata_negctrl = ndata[negctrl_idx]
-            param_history_dict["negctrl"] = run_inference(
-                negctrl_model, negctrl_guide, ndata_negctrl, num_steps=args.n_iter
+            param_history_dict["negctrl"] = deepcopy(
+                run_inference(
+                    negctrl_model, negctrl_guide, ndata_negctrl, num_steps=args.n_iter
+                )
             )
 
     outfile_path = (

diff --git a/bean/model/run.py b/bean/model/run.py
@@ -189,7 +189,7 @@ def run_inference(
         )
     return {
         "loss": losses,
-        "params": pyro.get_param_store(),
+        "params": pyro.get_param_store().get_state(),
     }