diff --git a/dev/registry/extract_metadata.py b/dev/registry/extract_metadata.py index 19b32dcbf18a0..0acabf69ece99 100644 --- a/dev/registry/extract_metadata.py +++ b/dev/registry/extract_metadata.py @@ -36,16 +36,17 @@ import re import shutil import subprocess +import sys import urllib.request import zlib from dataclasses import asdict, dataclass, field from pathlib import Path from typing import Any -try: +if sys.version_info >= (3, 11): import tomllib # Python 3.11+ stdlib -except ModuleNotFoundError: # pragma: no cover -- Python 3.10 fallback - import tomli as tomllib # type: ignore[no-redef] +else: # pragma: no cover -- Python 3.10 fallback + import tomli as tomllib import yaml from registry_contract_models import validate_providers_catalog diff --git a/dev/registry/extract_versions.py b/dev/registry/extract_versions.py index 2908b22b32e6a..cc5f094c79f05 100644 --- a/dev/registry/extract_versions.py +++ b/dev/registry/extract_versions.py @@ -46,10 +46,10 @@ from pathlib import Path from typing import Any -try: +if sys.version_info >= (3, 11): import tomllib # Python 3.11+ stdlib -except ModuleNotFoundError: # pragma: no cover -- Python 3.10 fallback - import tomli as tomllib # type: ignore[no-redef] +else: # pragma: no cover -- Python 3.10 fallback + import tomli as tomllib from registry_contract_models import validate_provider_version_metadata try: