diff --git a/torch_frame/__init__.py b/torch_frame/__init__.py index aa3dce17..182e0dbf 100644 --- 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, ])