.

pyg-team · Sep 9, 2024 · 4e4c3ea · 4e4c3ea
1 parent 9bb0585
commit 4e4c3ea
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/torch_frame/__init__.py b/torch_frame/__init__.py
@@ -27,13 +27,7 @@
 
 if WITH_PT24:
     import torch
-    import numpy as np
-    import codecs
     torch.serialization.add_safe_globals([
-        np._core.multiarray.scalar,
-        np.dtype,
-        np.dtypes.Int32DType,
-        codecs.encode,
         stype,
         torch_frame.data.stats.StatType,
     ])