From 0b162ff14e5564ca4afab73e22267772c4bb63f8 Mon Sep 17 00:00:00 2001 From: TATP-233 Date: Fri, 2 Oct 2026 05:39:20 +0800 Subject: [PATCH] perf(env): remove redundant state finite reduction --- src/unilab/base/torch_env.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/unilab/base/torch_env.py b/src/unilab/base/torch_env.py index a2f414f8a..cdc2b4a94 100644 --- a/src/unilab/base/torch_env.py +++ b/src/unilab/base/torch_env.py @@ -407,10 +407,8 @@ def _validate_state(self, state: TorchEnvState) -> None: self._validate_vector(steps, "info['steps']", torch.int64) self._validate_final_observation(state.final_observation) if finite_checks: - finite = torch.stack([torch.isfinite(value).all() for value, _ in finite_checks]) - if not bool(finite.all()): - for value, label in finite_checks: - self._validate_finite_float(value, label) + for value, label in finite_checks: + self._validate_finite_float(value, label) def _validate_vector(self, value: Any, label: str, dtype: torch.dtype) -> None: if not isinstance(value, torch.Tensor):