Skip to content

Commit 6ce1926

Browse files
committed
Refactor entry point utility functions
1 parent ac07f00 commit 6ce1926

File tree

1 file changed

+12
-12
lines changed

1 file changed

+12
-12
lines changed

asreviewcontrib/preprocess/utils.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -28,51 +28,51 @@ def get_entry_points(entry_name="asreview.preprocess.entry_points"):
2828
return {entry.name: entry for entry in pkg_resources.iter_entry_points(entry_name)}
2929

3030

31-
def _updater_class_from_entry_point(method, entry_name="asreview.preprocess.updaters"):
31+
def _updater_class_from_entry_point(updater, entry_name="asreview.preprocess.updaters"):
3232
entry_points = get_entry_points(entry_name)
3333
try:
34-
return entry_points[method].load()
34+
return entry_points[updater].load()
3535
except KeyError:
3636
raise ValueError(
37-
f"Error: method '{method}' is not implemented for entry point "
37+
f"Error: updater '{updater}' is not implemented for entry point "
3838
f"{entry_name}."
3939
)
4040
except ImportError as e:
4141
raise ValueError(
42-
f"Failed to import '{method}' updater ({entry_name}) "
42+
f"Failed to import '{updater}' updater ({entry_name}) "
4343
f"with the following error:\n{e}"
4444
)
4545

4646

47-
def _localdb_class_from_entry_point(suffix, entry_name="asreview.preprocess.localdbs"):
47+
def _localdb_class_from_entry_point(localdb, entry_name="asreview.preprocess.localdbs"):
4848
entry_points = get_entry_points(entry_name)
4949
try:
50-
return entry_points[suffix].load()
50+
return entry_points[localdb].load()
5151
except KeyError:
5252
raise ValueError(
53-
f"Error: suffix '{suffix}' is not implemented for entry point "
53+
f"Error: localdb '{localdb}' is not implemented for entry point "
5454
f"{entry_name}."
5555
)
5656
except ImportError as e:
5757
raise ValueError(
58-
f"Failed to import '{suffix}' localdb ({entry_name}) "
58+
f"Failed to import '{localdb}' localdb ({entry_name}) "
5959
f"with the following error:\n{e}"
6060
)
6161

6262

6363
def _deduplicator_class_from_entry_point(
64-
suffix, entry_name="asreview.preprocess.deduplicators"
64+
deduplicator, entry_name="asreview.preprocess.deduplicators"
6565
):
6666
entry_points = get_entry_points(entry_name)
6767
try:
68-
return entry_points[suffix].load()
68+
return entry_points[deduplicator].load()
6969
except KeyError:
7070
raise ValueError(
71-
f"Error: suffix '{suffix}' is not implemented for entry point "
71+
f"Error: deduplicator '{deduplicator}' is not implemented for entry point "
7272
f"{entry_name}."
7373
)
7474
except ImportError as e:
7575
raise ValueError(
76-
f"Failed to import '{suffix}' deduplicator ({entry_name}) "
76+
f"Failed to import '{deduplicator}' deduplicator ({entry_name}) "
7777
f"with the following error:\n{e}"
7878
)

0 commit comments

Comments
 (0)