From 9578c2597e2d88b6f0b304b5a05864fd613ddcc1 Mon Sep 17 00:00:00 2001 From: Zach Mueller Date: Mon, 26 Aug 2024 12:27:39 -0400 Subject: [PATCH] Fixup py 38 type hints for mps friendly (#33128) Fixup py 38 --- src/transformers/pytorch_utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/transformers/pytorch_utils.py b/src/transformers/pytorch_utils.py index 8c1bd21fb2962a..f3663c09902f52 100644 --- a/src/transformers/pytorch_utils.py +++ b/src/transformers/pytorch_utils.py @@ -11,6 +11,8 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. +from __future__ import annotations + import inspect from typing import Callable, List, Optional, Set, Tuple, Union