fix(nyz): fix zergling collector unittest bug

opendilab · Mar 29, 2023 · e873d9e · e873d9e
1 parent afcfa51
commit e873d9e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ding/worker/collector/zergling_parallel_collector.py b/ding/worker/collector/zergling_parallel_collector.py
@@ -80,8 +80,8 @@ def policy(self, _policy: Policy) -> None:
         self._policy = _policy
         self._policy_cfg = self._policy.get_attribute('cfg')
         self._n_sample = _policy.get_attribute('n_sample')
-        self._n_episode = _policy.get_attribute('n_sample')
-        assert any(
+        self._n_episode = _policy.get_attribute('n_episode')
+        assert not all(
             [t is None for t in [self._n_sample, self._n_episode]]
         ), "n_episode/n_sample in policy cfg can't be not None at the same time"
         # TODO(nyz) the same definition of traj_len in serial and parallel