From 344e06d012a1f246a8f9993d91c5433daf336a15 Mon Sep 17 00:00:00 2001 From: thatguy11325 <148832074+thatguy11325@users.noreply.github.com> Date: Sun, 11 Aug 2024 22:03:02 -0400 Subject: [PATCH] not is import --- pokemonred_puffer/policies/multi_convolutional.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pokemonred_puffer/policies/multi_convolutional.py b/pokemonred_puffer/policies/multi_convolutional.py index 22f75b2..7636a51 100644 --- 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,