⚠️ Time to say goodbye to py37 (#24091)
* fix --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
@@ -13,6 +13,7 @@
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
import gc
|
||||
import importlib.metadata
|
||||
import tempfile
|
||||
import unittest
|
||||
|
||||
@@ -36,7 +37,6 @@ from transformers.testing_utils import (
|
||||
require_torch_multi_gpu,
|
||||
slow,
|
||||
)
|
||||
from transformers.utils.versions import importlib_metadata
|
||||
|
||||
|
||||
def get_some_linear_layer(model):
|
||||
@@ -446,7 +446,7 @@ class Bnb4BitTestTraining(Base4bitTest):
|
||||
super().setUp()
|
||||
|
||||
def test_training(self):
|
||||
if version.parse(importlib_metadata.version("bitsandbytes")) < version.parse("0.37.0"):
|
||||
if version.parse(importlib.metadata.version("bitsandbytes")) < version.parse("0.37.0"):
|
||||
return
|
||||
|
||||
# Step 1: freeze all parameters
|
||||
|
||||
@@ -13,6 +13,7 @@
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
import gc
|
||||
import importlib.metadata
|
||||
import tempfile
|
||||
import unittest
|
||||
|
||||
@@ -38,7 +39,6 @@ from transformers.testing_utils import (
|
||||
require_torch_multi_gpu,
|
||||
slow,
|
||||
)
|
||||
from transformers.utils.versions import importlib_metadata
|
||||
|
||||
|
||||
def get_some_linear_layer(model):
|
||||
@@ -722,7 +722,7 @@ class MixedInt8TestTraining(BaseMixedInt8Test):
|
||||
super().setUp()
|
||||
|
||||
def test_training(self):
|
||||
if version.parse(importlib_metadata.version("bitsandbytes")) < version.parse("0.37.0"):
|
||||
if version.parse(importlib.metadata.version("bitsandbytes")) < version.parse("0.37.0"):
|
||||
return
|
||||
|
||||
# Step 1: freeze all parameters
|
||||
|
||||
@@ -22,7 +22,7 @@ from argparse import Namespace
|
||||
from dataclasses import dataclass, field
|
||||
from enum import Enum
|
||||
from pathlib import Path
|
||||
from typing import List, Optional
|
||||
from typing import List, Literal, Optional
|
||||
|
||||
import yaml
|
||||
|
||||
@@ -30,13 +30,6 @@ from transformers import HfArgumentParser, TrainingArguments
|
||||
from transformers.hf_argparser import make_choice_type_function, string_to_bool
|
||||
|
||||
|
||||
try:
|
||||
# For Python versions <3.8, Literal is not in typing: https://peps.python.org/pep-0586/
|
||||
from typing import Literal
|
||||
except ImportError:
|
||||
# For Python 3.7
|
||||
from typing_extensions import Literal
|
||||
|
||||
# Since Python 3.10, we can use the builtin `|` operator for Union types
|
||||
# See PEP 604: https://peps.python.org/pep-0604
|
||||
is_python_no_less_than_3_10 = sys.version_info >= (3, 10)
|
||||
|
||||
@@ -12,13 +12,14 @@
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
import importlib.metadata
|
||||
import sys
|
||||
|
||||
from transformers.testing_utils import TestCasePlus
|
||||
from transformers.utils.versions import importlib_metadata, require_version, require_version_core
|
||||
from transformers.utils.versions import require_version, require_version_core
|
||||
|
||||
|
||||
numpy_ver = importlib_metadata.version("numpy")
|
||||
numpy_ver = importlib.metadata.version("numpy")
|
||||
python_ver = ".".join([str(x) for x in sys.version_info[:3]])
|
||||
|
||||
|
||||
@@ -65,7 +66,7 @@ class DependencyVersionCheckTest(TestCasePlus):
|
||||
for req in ["numpipypie>1", "numpipypie2"]:
|
||||
try:
|
||||
require_version_core(req)
|
||||
except importlib_metadata.PackageNotFoundError as e:
|
||||
except importlib.metadata.PackageNotFoundError as e:
|
||||
self.assertIn(f"The '{req}' distribution was not found and is required by this application", str(e))
|
||||
self.assertIn("Try: pip install transformers -U", str(e))
|
||||
|
||||
|
||||
Reference in New Issue
Block a user