Fix tuple type hint

huggingface · Aug 17, 2023 · 27b0d60 · 27b0d60
1 parent c344cec
commit 27b0d60
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/optimum/exporters/ggml/base.py b/optimum/exporters/ggml/base.py
@@ -15,7 +15,7 @@
 """ggml configuration base classes."""
 
 from abc import ABC, abstractmethod
-from typing import Dict, Union
+from typing import Dict, Tuple, Union
 
 from numpy import ndarray
 from torch import Tensor
@@ -49,7 +49,7 @@ def reshape_weights(self, name: str, weights: Union[ndarray, Tensor], hparams: D
 
     @staticmethod
     @abstractmethod
-    def convert_dtype(name: str, data: Union[ndarray, Tensor], ftype: int, n_dims: int) -> tuple[ndarray, int]:
+    def convert_dtype(name: str, data: Union[ndarray, Tensor], ftype: int, n_dims: int) -> Tuple[ndarray, int]:
         return data, ftype