WIP: better initialization of agent

Author: smejak

blockference/agent.py

@@ -48,6 +48,33 @@ def __init__(
         lr_pD=1.0,
         use_BMA=True,
         policy_sep_prior=False,
-        save_belief_hist=False
+        save_belief_hist=False,
     ):
-        super().__init__()
+        super().__init__(A,
+                         B,
+                         C=None,
+                         D=None,
+                         E=None,
+                         pA=None,
+                         pB=None,
+                         pD=None,
+                         num_controls=None,
+                         policy_len=1,
+                         inference_horizon=1,
+                         control_fac_idx=None,
+                         policies=None,
+                         gamma=16.0,
+                         use_utility=True,
+                         use_states_info_gain=True,
+                         use_param_info_gain=False,
+                         action_selection="deterministic",
+                         inference_algo="VANILLA",
+                         inference_params=None,
+                         modalities_to_learn="all",
+                         lr_pA=1.0,
+                         factors_to_learn="all",
+                         lr_pB=1.0,
+                         lr_pD=1.0,
+                         use_BMA=True,
+                         policy_sep_prior=False,
+                         save_belief_hist=False,)