diff --git a/ml-agents/mlagents/trainers/policy/policy.py b/ml-agents/mlagents/trainers/policy/policy.py index e6cd9fedbd..b4de537253 100644 --- a/ml-agents/mlagents/trainers/policy/policy.py +++ b/ml-agents/mlagents/trainers/policy/policy.py @@ -140,10 +140,6 @@ def check_nan_action(action: Optional[ActionTuple]) -> None: has_nan = np.isnan(d) if has_nan: raise RuntimeError("Continuous NaN action detected.") - d = np.sum(action.discrete) - has_nan = np.isnan(d) - if has_nan: - raise RuntimeError("Discrete NaN action detected.") @abstractmethod def update_normalization(self, vector_obs: np.ndarray) -> None: