From 1bb19b3fea9773abdee6309b21fa59d145c0f410 Mon Sep 17 00:00:00 2001 From: Chris Elion Date: Thu, 17 Dec 2020 15:12:13 -0800 Subject: [PATCH] remove "discrete nan" check (#4770) --- ml-agents/mlagents/trainers/policy/policy.py | 4 ---- 1 file changed, 4 deletions(-) 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: