Skip to content

Commit d379032

Browse files
committed
fix: remove non existent types
1 parent e8df097 commit d379032

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

binding/python/py/bintensors/numpy.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -178,9 +178,6 @@ def load_file(filename: Union[str, os.PathLike]) -> Dict[str, np.ndarray]:
178178

179179

180180
# np.float8 formats require 2.1; we do not support these dtypes on earlier versions
181-
_float8_e4m3fn = getattr(np, "float8_e4m3fn", None)
182-
_float8_e5m2 = getattr(np, "float8_e5m2", None)
183-
184181
_TYPES = {
185182
"F64": np.float64,
186183
"F32": np.float32,
@@ -193,9 +190,7 @@ def load_file(filename: Union[str, os.PathLike]) -> Dict[str, np.ndarray]:
193190
"U16": np.uint16,
194191
"I8": np.int8,
195192
"U8": np.uint8,
196-
"BOOL": np.bool,
197-
"F8_E4M3": _float8_e4m3fn,
198-
"F8_E5M2": _float8_e5m2,
193+
"BOOL": bool,
199194
}
200195

201196

0 commit comments

Comments
 (0)