Skip to content

Commit

Permalink
fix DiscretePolicyGradient set_state bug; upgrade pymaro version
Browse files Browse the repository at this point in the history
  • Loading branch information
Jinyu Wang committed May 15, 2023
1 parent d21fb40 commit b3c6a58
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion maro/__misc__.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,6 @@
# Licensed under the MIT license.


__version__ = "0.3.2a2"
__version__ = "0.3.2a3"

__data_version__ = "0.2"
2 changes: 1 addition & 1 deletion maro/rl/policy/discrete_rl_policy.py
Original file line number Diff line number Diff line change
Expand Up @@ -336,7 +336,7 @@ def get_state(self) -> dict:
}

def set_state(self, policy_state: dict) -> None:
self._policy_net.set_state(policy_state)
self._policy_net.set_state(policy_state["net"])
self._warmup = policy_state["policy"]["warmup"]
self._call_count = policy_state["policy"]["call_count"]

Expand Down

0 comments on commit b3c6a58

Please sign in to comment.