not is import

thatguy11325 · Aug 12, 2024 · 344e06d · 344e06d
1 parent 89b2582
commit 344e06d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pokemonred_puffer/policies/multi_convolutional.py b/pokemonred_puffer/policies/multi_convolutional.py
@@ -232,7 +232,7 @@ def encode_observations(self, observations):
             ),
             dim=-1,
         )
-        if self.skip_safari_zone:
+        if not self.skip_safari_zone:
             cat_obs = torch.cat(
                 (
                     cat_obs,