Skip to content

Commit 54b4746

Browse files
#2498 try different entry points method
1 parent 4f233b0 commit 54b4746

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

pybamm/parameters/parameter_sets.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import warnings
2-
import pkg_resources
2+
import entrypoints
33
import textwrap
44
from collections.abc import Mapping
55

@@ -36,7 +36,7 @@ class ParameterSets(Mapping):
3636
def __init__(self):
3737
# Dict of entry points for parameter sets, lazily load entry points as
3838
self.__all_parameter_sets = dict()
39-
for entry_point in pkg_resources.iter_entry_points("pybamm_parameter_sets"):
39+
for entry_point in entrypoints.get_group_all("pybamm_parameter_sets"):
4040
self.__all_parameter_sets[entry_point.name] = entry_point
4141

4242
def __new__(cls):
@@ -55,7 +55,7 @@ def __load_entry_point__(self, key) -> callable:
5555
if key not in self.__all_parameter_sets:
5656
raise KeyError(f"Unknown parameter set: {key}")
5757
ps = self.__all_parameter_sets[key]
58-
if isinstance(ps, pkg_resources.EntryPoint):
58+
if isinstance(ps, entrypoints.EntryPoint):
5959
ps = self.__all_parameter_sets[key] = ps.load()
6060
return ps
6161

0 commit comments

Comments
 (0)