Skip to content
Open
Show file tree
Hide file tree
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
1 change: 1 addition & 0 deletions _typos.toml
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@ extend-ignore-identifiers-re = [
# Add specific words that should not be corrected
# rhapsody = "rhapsody"
HPE = "HPE"
leafs = "leafs"

[files]
extend-exclude = [
Expand Down
8 changes: 8 additions & 0 deletions src/rhapsody/backends/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,6 +59,14 @@
except ImportError:
pass

try:
from .execution import EnsembleExecutionBackend

__all__.append(EnsembleExecutionBackend)

except ImportError:
pass

# Try to import optional inference backends
try:
from .inference.vllm import DragonVllmInferenceBackend # noqa: F401
Expand Down
7 changes: 7 additions & 0 deletions src/rhapsody/backends/execution/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,13 @@
except ImportError:
pass

try:
from .el import EnsembleExecutionBackend # noqa: F401

__all__.append("EnsembleExecutionBackend")
except ImportError:
pass

try:
from .dragon import DragonExecutionBackendV1 # noqa: F401
from .dragon import DragonExecutionBackendV2 # noqa: F401
Expand Down
Loading