From 4046ea923fb760d3d5174873f5fd471e4ac53b36 Mon Sep 17 00:00:00 2001
From: thatguy11325 <148832074+thatguy11325@users.noreply.github.com>
Date: Thu, 11 Jul 2024 07:02:49 -0400
Subject: [PATCH] compile disable nativize tensor

---
 pokemonred_puffer/policies/multi_convolutional.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pokemonred_puffer/policies/multi_convolutional.py b/pokemonred_puffer/policies/multi_convolutional.py
index c38ac63..8ea5b97 100644
--- a/pokemonred_puffer/policies/multi_convolutional.py
+++ b/pokemonred_puffer/policies/multi_convolutional.py
@@ -8,6 +8,8 @@
 from pokemonred_puffer.data.items import Items
 from pokemonred_puffer.environment import PIXEL_VALUES
 
+pufferlib.pytorch.nativize_tensor = torch.compiler.disable(pufferlib.pytorch.nativize_tensor)
+
 
 # Because torch.nn.functional.one_hot cannot be traced by torch as of 2.2.0
 def one_hot(tensor, num_classes):