From: compilade <redacted>
Date: Tue, 9 Jul 2024 05:04:49 +0000 (-0400)
Subject: gguf-py : do not use internal numpy types (#7472)
X-Git-Tag: upstream/0.0.4488~1136
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=7d0e23d72ef4540d0d4409cb63ae682c17d53926;p=pkg%2Fggml%2Fsources%2Fllama.cpp

gguf-py : do not use internal numpy types (#7472)
---

diff --git a/gguf-py/gguf/lazy.py b/gguf-py/gguf/lazy.py
index c50124cd..6e266f34 100644
--- a/gguf-py/gguf/lazy.py
+++ b/gguf-py/gguf/lazy.py
@@ -6,7 +6,6 @@ from typing import Any, Callable
 from collections import deque
 
 import numpy as np
-from numpy._typing import _Shape
 from numpy.typing import DTypeLike
 
 
@@ -219,7 +218,7 @@ class LazyNumpyTensor(LazyBase):
     _tensor_type = np.ndarray
 
     @classmethod
-    def meta_with_dtype_and_shape(cls, dtype: DTypeLike, shape: _Shape) -> np.ndarray[Any, Any]:
+    def meta_with_dtype_and_shape(cls, dtype: DTypeLike, shape: tuple[int, ...]) -> np.ndarray[Any, Any]:
         # The initial idea was to use np.nan as the fill value,
         # but non-float types like np.int16 can't use that.
         # So zero it is.