[CI] Phase out Python 3.8 wheel builds (#633)

tqchen · web-flow · commit fcd5ab4f5bd3 · 2026-06-18T14:04:41.000-04:00
cibuildwheel 4.x no longer accepts Python 3.8 build selectors, so the
wheel build configuration needs to start at Python 3.9.

This PR:
- raises the package Python floor to 3.9
- removes cp38 from cibuildwheel build and test-skip selectors
- updates the CI test matrix Python 3.8 leg to Python 3.9
diff --git a/.github/actions/build-wheel-for-publish/action.yml b/.github/actions/build-wheel-for-publish/action.yml
@@ -45,14 +45,6 @@ inputs:
 runs:
   using: "composite"
   steps:
-    # Special handling for macOS arm64 + python 3.8.
-    # Install Python 3.8 from `actions/setup-python`, which is an arm64 build.
-    - name: Install Python 3.8 for macOS arm64
-      if: runner.os == 'macOS' && inputs.arch == 'arm64'
-      uses: actions/setup-python@v5
-      with:
-        python-version: 3.8
-
     - name: Set up uv
       uses: astral-sh/setup-uv@fac544c07dec837d0ccb6301d7b5580bf5edae39  # v8.2.0
 
diff --git a/.github/workflows/ci_test.yml b/.github/workflows/ci_test.yml
@@ -131,7 +131,7 @@ jobs:
       matrix:
         include:
           - {os: ubuntu-latest, arch: x86_64, python_version: '3.14t'}
-          - {os: ubuntu-24.04-arm, arch: aarch64, python_version: '3.8'}
+          - {os: ubuntu-24.04-arm, arch: aarch64, python_version: '3.14'}
           - {os: windows-latest, arch: AMD64, python_version: '3.9'}
           - {os: macos-14, arch: arm64, python_version: '3.13'}
 
diff --git a/pyproject.toml b/pyproject.toml
@@ -31,7 +31,7 @@ classifiers = [
   "Intended Audience :: Science/Research",
 ]
 keywords = ["machine learning", "inference"]
-requires-python = ">=3.8"
+requires-python = ">=3.9"
 dependencies = ["typing-extensions>=4.5"]
 
 [project.urls]
@@ -181,7 +181,7 @@ testpaths = ["tests"]
 include = ["python/**/*.py", "tests/**/*.py"]
 line-length = 100
 indent-width = 4
-target-version = "py38"
+target-version = "py39"
 
 [tool.ruff.lint]
 select = [
@@ -235,10 +235,10 @@ build-verbosity = 1
 # only build up to cp312, cp312
 # will be abi3 and can be used in future versions
 # ship 314t threaded nogil version
-build = ["cp38-*", "cp39-*", "cp310-*", "cp311-*", "cp312-*", "cp314t-*"]
+build = ["cp39-*", "cp310-*", "cp311-*", "cp312-*", "cp314t-*"]
 skip = ["*musllinux*"]
 # we only need to test on cp312
-test-skip = ["cp38-*", "cp39-*", "cp310-*", "cp311-*"]
+test-skip = ["cp39-*", "cp310-*", "cp311-*"]
 # focus on testing abi3 wheel
 build-frontend = "build[uv]"
 test-command = "pytest {package}/tests/python -vvs"
@@ -267,6 +267,7 @@ unused-ignore-comment = "ignore"
 
 [tool.ty.analysis]
 allowed-unresolved-imports = [
+  "tvm_ffi._version",
   "torch",
   "torch.*",
   "torch.utils.*",
diff --git a/python/tvm_ffi/container.py b/python/tvm_ffi/container.py
@@ -20,7 +20,17 @@
 
 import itertools
 import operator
-import sys
+from collections.abc import ItemsView as ItemsViewBase
+from collections.abc import (
+    Iterable,
+    Iterator,
+    Mapping,
+    MutableMapping,
+    MutableSequence,
+    Sequence,
+)
+from collections.abc import KeysView as KeysViewBase
+from collections.abc import ValuesView as ValuesViewBase
 from typing import (
     Any,
     Callable,
@@ -33,46 +43,6 @@
 from . import _ffi_api, core
 from .registry import register_object
 
-if sys.version_info >= (3, 9):
-    # PEP 585 generics
-    from collections.abc import (
-        ItemsView as ItemsViewBase,
-    )
-    from collections.abc import (
-        Iterable,
-        Iterator,
-        Mapping,
-        MutableMapping,
-        MutableSequence,
-        Sequence,
-    )
-    from collections.abc import (
-        KeysView as KeysViewBase,
-    )
-    from collections.abc import (
-        ValuesView as ValuesViewBase,
-    )
-else:  # Python 3.8
-    # workarounds for python 3.8
-    # typing-module generics (subscriptable on 3.8)
-    from typing import (
-        ItemsView as ItemsViewBase,
-    )
-    from typing import (
-        Iterable,
-        Iterator,
-        Mapping,
-        MutableMapping,
-        MutableSequence,
-        Sequence,
-    )
-    from typing import (
-        KeysView as KeysViewBase,
-    )
-    from typing import (
-        ValuesView as ValuesViewBase,
-    )
-
 __all__ = ["Array", "Dict", "List", "Map"]
 
 
diff --git a/python/tvm_ffi/cpp/dtype.py b/python/tvm_ffi/cpp/dtype.py
@@ -59,7 +59,7 @@
 }
 
 
-@functools.lru_cache(maxsize=None)
+@functools.cache
 def _determine_backend_once() -> Literal["cpu", "cuda", "rocm"]:
     try:
         import torch  # noqa: PLC0415
diff --git a/python/tvm_ffi/cpp/nvrtc.py b/python/tvm_ffi/cpp/nvrtc.py
@@ -18,7 +18,7 @@
 
 from __future__ import annotations
 
-from typing import Sequence
+from collections.abc import Sequence
 
 
 def nvrtc_compile(  # noqa: PLR0912, PLR0915
diff --git a/python/tvm_ffi/registry.py b/python/tvm_ffi/registry.py
@@ -21,7 +21,8 @@
 import json
 import sys
 import warnings
-from typing import Any, Callable, Literal, Sequence, TypeVar, overload
+from collections.abc import Sequence
+from typing import Any, Callable, Literal, TypeVar, overload
 
 from . import core
 from .core import Function, TypeInfo
diff --git a/python/tvm_ffi/stub/file_utils.py b/python/tvm_ffi/stub/file_utils.py
@@ -22,8 +22,9 @@
 import difflib
 import os
 import traceback
+from collections.abc import Generator, Iterable
 from pathlib import Path
-from typing import Callable, Generator, Iterable
+from typing import Callable
 
 from . import consts as C
 
diff --git a/python/tvm_ffi/stub/lib_state.py b/python/tvm_ffi/stub/lib_state.py
@@ -30,7 +30,7 @@
 from .utils import FuncInfo, NamedTypeSchema, ObjectInfo
 
 
-@functools.lru_cache(maxsize=None)
+@functools.cache
 def object_info_from_type_key(type_key: str) -> ObjectInfo:
     """Construct an `ObjectInfo` from an object type key."""
     type_info = _lookup_or_register_type_info_from_type_key(str(type_key))
@@ -108,7 +108,7 @@ def toposort_objects(type_keys: list[str]) -> list[ObjectInfo]:
     return [infos[type_key] for type_key in sorted_keys]
 
 
-@functools.lru_cache(maxsize=None)
+@functools.cache
 def _func_info_from_global_name(name: str) -> FuncInfo:
     """Construct a `FuncInfo` from a global function name."""
     return FuncInfo(
diff --git a/python/tvm_ffi/testing/testing.py b/python/tvm_ffi/testing/testing.py
@@ -41,10 +41,6 @@
 from .. import _ffi_api
 from .. import core as tvm_ffi_core
 
-requires_py39 = pytest.mark.skipif(
-    sys.version_info < (3, 9),
-    reason="requires Python 3.9+",
-)
 requires_py310 = pytest.mark.skipif(
     sys.version_info < (3, 10),
     reason="requires Python 3.10+",
diff --git a/python/tvm_ffi/utils/kwargs_wrapper.py b/python/tvm_ffi/utils/kwargs_wrapper.py
@@ -25,7 +25,8 @@
 import functools
 import inspect
 import keyword
-from typing import Any, Callable, Iterable
+from collections.abc import Iterable
+from typing import Any, Callable
 
 from tvm_ffi.utils.unpack_dataclass import unpack_dataclass_to_tuple
 
diff --git a/tests/python/test_container.py b/tests/python/test_container.py
@@ -17,7 +17,7 @@
 from __future__ import annotations
 
 import pickle
-import sys
+from collections.abc import Sequence
 from typing import Any
 
 import pytest
@@ -26,14 +26,6 @@
 from tvm_ffi.container import MISSING as CONTAINER_MISSING
 from tvm_ffi.core import MISSING
 
-if sys.version_info >= (3, 9):
-    # PEP 585 generics
-    from collections.abc import Sequence
-else:  # Python 3.8
-    # workarounds for python 3.8
-    # typing-module generics (subscriptable on 3.8)
-    from typing import Sequence
-
 
 def test_array() -> None:
     a = tvm_ffi.convert([1, 2, 3])
diff --git a/tests/python/test_dataclass_common.py b/tests/python/test_dataclass_common.py
@@ -14,7 +14,7 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-# ruff: noqa: D102, UP006, UP045
+# ruff: noqa: D102, UP006, UP035, UP045
 """Tests for :func:`tvm_ffi.dataclasses.is_dataclass`, :func:`fields`, :func:`replace`."""
 
 from __future__ import annotations
diff --git a/tests/python/test_dataclass_copy.py b/tests/python/test_dataclass_copy.py
@@ -14,7 +14,7 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-# ruff: noqa: D102, UP006, UP045
+# ruff: noqa: D102, UP006, UP035, UP045
 """Tests for __copy__, __deepcopy__, and __replace__ on FFI objects."""
 
 from __future__ import annotations
diff --git a/tests/python/test_dataclass_frozen.py b/tests/python/test_dataclass_frozen.py
@@ -16,7 +16,7 @@
 # under the License.
 """Tests for frozen support in ``@py_class``."""
 
-# ruff: noqa: D102
+# ruff: noqa: D102, UP006, UP035, UP045
 from __future__ import annotations
 
 import copy
@@ -95,7 +95,7 @@ class TestFrozenFieldTypes:
     def test_frozen_optional_field(self) -> None:
         @py_class(_unique_key("opt_field"), frozen=True)
         class Opt(Object):
-            v: Optional[int]  # noqa: UP045
+            v: Optional[int]
 
         o1 = Opt(None)
         assert o1.v is None
@@ -122,7 +122,7 @@ class Outer(Object):
     def test_frozen_list_field(self) -> None:
         @py_class(_unique_key("list_field"), frozen=True)
         class HasList(Object):
-            items: List[int]  # noqa: UP006
+            items: List[int]
 
         h = HasList([1, 2, 3])
         assert len(h.items) == 3
@@ -132,7 +132,7 @@ class HasList(Object):
     def test_frozen_dict_field(self) -> None:
         @py_class(_unique_key("dict_field"), frozen=True)
         class HasDict(Object):
-            mapping: Dict[str, int]  # noqa: UP006
+            mapping: Dict[str, int]
 
         h = HasDict({"a": 1})
         assert h.mapping["a"] == 1
@@ -157,7 +157,7 @@ class Cfg(Object):
     def test_frozen_field_with_default_factory(self) -> None:
         @py_class(_unique_key("def_factory"), frozen=True)
         class Cfg(Object):
-            items: List[int] = field(default_factory=list)  # noqa: UP006
+            items: List[int] = field(default_factory=list)
 
         c = Cfg()
         assert len(c.items) == 0
@@ -252,7 +252,7 @@ class Pt(Object):
     def test_escape_hatch_preserves_type_coercion(self) -> None:
         @py_class(_unique_key("esc_coerce"), frozen=True)
         class HasList(Object):
-            items: List[int]  # noqa: UP006
+            items: List[int]
 
         h = HasList([1])
         type(h).items.set(h, [10, 20])  # ty: ignore[unresolved-attribute]
@@ -610,7 +610,7 @@ class Single(Object):
     def test_frozen_field_with_none_default(self) -> None:
         @py_class(_unique_key("none_def"), frozen=True)
         class Opt(Object):
-            v: Optional[int] = None  # noqa: UP045
+            v: Optional[int] = None
 
         o = Opt()
         assert o.v is None
diff --git a/tests/python/test_dataclass_py_class.py b/tests/python/test_dataclass_py_class.py
@@ -16,7 +16,7 @@
 # under the License.
 """Tests for Python-defined TVM-FFI types: ``@py_class`` decorator and low-level Field API."""
 
-# ruff: noqa: D102, PLR0124, PLW1641, UP006, UP045
+# ruff: noqa: D102, PLR0124, PLW1641, UP006, UP035, UP045
 from __future__ import annotations
 
 import copy
diff --git a/tests/python/test_object.py b/tests/python/test_object.py
@@ -17,7 +17,8 @@
 from __future__ import annotations
 
 import sys
-from typing import Any, Sequence
+from collections.abc import Sequence
+from typing import Any
 
 import pytest
 import tvm_ffi
diff --git a/tests/python/test_type_converter.py b/tests/python/test_type_converter.py

Original file line number	Diff line number	Diff line change
`@@ -59,7 +59,7 @@`
`59`	`59`	`}`
`60`	`60`
`61`	`61`
`62`		`-@functools.lru_cache(maxsize=None)`
	`62`	`+@functools.cache`
`63`	`63`	`def _determine_backend_once() -> Literal["cpu", "cuda", "rocm"]:`
`64`	`64`	`try:`
`65`	`65`	`import torch # noqa: PLC0415`