diff --git a/declearn/test_utils/_vectors.py b/declearn/test_utils/_vectors.py index e0f37b873848748fc87b22b7f23f62d2cf9361ed..dbc6349128703bc040c0e9458245bd98d749d2e1 100644 --- a/declearn/test_utils/_vectors.py +++ b/declearn/test_utils/_vectors.py @@ -18,11 +18,11 @@ """Shared objects for testing purposes.""" import importlib +import importlib.metadata import typing from typing import List, Literal, Optional, Type import numpy as np -import pkg_resources # type: ignore from numpy.typing import ArrayLike from declearn.model.api import Vector @@ -43,8 +43,8 @@ def list_available_frameworks() -> List[FrameworkType]: available = [] for framework in typing.get_args(FrameworkType): try: - pkg_resources.require(framework) - except pkg_resources.DistributionNotFound: + importlib.metadata.distribution(framework) + except importlib.metadata.PackageNotFoundError: pass else: available.append(framework)