fix DiscreteRLPolicy set_state bug

Jinyu Wang · Jinyu-W · commit a9a638134bef · 2023-05-15T12:02:01.000+08:00
diff --git a/maro/rl/policy/discrete_rl_policy.py b/maro/rl/policy/discrete_rl_policy.py
@@ -244,7 +244,7 @@ def get_state(self) -> dict:
         }
 
     def set_state(self, policy_state: dict) -> None:
-        self._q_net.set_state(policy_state)
+        self._q_net.set_state(policy_state["net"])
         self._warmup = policy_state["policy"]["warmup"]
         self._call_count = policy_state["policy"]["call_count"]
 

Original file line number	Diff line number	Diff line change
`@@ -244,7 +244,7 @@ def get_state(self) -> dict:`
`244`	`244`	`}`
`245`	`245`
`246`	`246`	`def set_state(self, policy_state: dict) -> None:`
`247`		`- self._q_net.set_state(policy_state)`
	`247`	`+ self._q_net.set_state(policy_state["net"])`
`248`	`248`	`self._warmup = policy_state["policy"]["warmup"]`
`249`	`249`	`self._call_count = policy_state["policy"]["call_count"]`
`250`	`250`