fix test

thu-ml · Apr 24, 2021 · 2c1cc1d · 2c1cc1d
1 parent 68ae216
commit 2c1cc1d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tianshou/policy/modelfree/trpo.py b/tianshou/policy/modelfree/trpo.py
@@ -122,7 +122,7 @@ def learn(  # type: ignore
                                           " are poor and need to be changed.")
 
                 # optimize citirc
-                for _ in range(self._optim_critic_iters):
+                for _ in range(self._optim_critic_iters):  # type: ignore
                     value = self.critic(b.obs).flatten()
                     vf_loss = F.mse_loss(b.returns, value)
                     self.optim.zero_grad()