Skip to content

Commit

Permalink
removed field(static=True) option for policies matrix in Agent
Browse files Browse the repository at this point in the history
  • Loading branch information
conorheins committed Sep 24, 2024
1 parent cc48366 commit d235ca6
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions pymdp/agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,6 +46,9 @@ class Agent(Module):
gamma: Array
alpha: Array

# matrix of all possible policies (each row is a policy of shape (num_controls[0], num_controls[1], ..., num_controls[num_control_factors-1])
policies: Array

# threshold for inductive inference (the threshold for pruning transitions that are below a certain probability)
inductive_threshold: Array
# epsilon for inductive inference (trade-off/weight for how much inductive value contributes to EFE of policies)
Expand Down Expand Up @@ -74,8 +77,6 @@ class Agent(Module):
policy_len: int = field(static=True)
# depth of inductive inference (i.e. number of future timesteps to use when computing inductive `I` matrix)
inductive_depth: int = field(static=True)
# matrix of all possible policies (each row is a policy of shape (num_controls[0], num_controls[1], ..., num_controls[num_control_factors-1])
policies: Array = field(static=True)
# flag for whether to use expected utility ("reward" or "preference satisfaction") when computing expected free energy
use_utility: bool = field(static=True)
# flag for whether to use state information gain ("salience") when computing expected free energy
Expand Down

0 comments on commit d235ca6

Please sign in to comment.