[RLlib] Atari-RAM-Preprocessing, unsigned observation vector results in a false preprocessed observation (#13013)

2026-06-27 18:06:25 +08:00 · 2021-01-28 12:07:00 +01:00
parent 56ee6ef55f
commit d4ef5c5993
1 changed files with 1 additions and 1 deletions
@@ -140,7 +140,7 @@ class AtariRamPreprocessor(Preprocessor):
    @override(Preprocessor)
    def transform(self, observation: TensorType) -> np.ndarray:
        self.check_shape(observation)
-        return (observation - 128) / 128
+        return (observation.astype("float32") - 128) / 128


 class OneHotPreprocessor(Preprocessor):