Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions src/experimaestro/launcherfinder/registry.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
from pathlib import Path
import typing
from omegaconf import DictConfig, OmegaConf, SCMode
import pkg_resources
from importlib.metadata import entry_points
from experimaestro.utils import logger
from .base import ConnectorConfiguration, TokenConfiguration
from .specs import HostRequirement, RequirementUnion
Expand Down Expand Up @@ -75,10 +75,10 @@ def __init__(self, basepath: Path):
self.find_launcher_fn = None

# Use entry points for connectors and launchers
for entry_point in pkg_resources.iter_entry_points("experimaestro.connectors"):
for entry_point in entry_points(group="experimaestro.connectors"):
entry_point.load().init_registry(self)

for entry_point in pkg_resources.iter_entry_points("experimaestro.tokens"):
for entry_point in entry_points(group="experimaestro.tokens"):
entry_point.load().init_registry(self)

# Register the find launcher function if it exists
Expand Down