Skip to content

Commit

Permalink
Remove ray static loading in compute build
Browse files Browse the repository at this point in the history
  • Loading branch information
kartik4949 committed Aug 19, 2024
1 parent a1f61e4 commit c04795c
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 10 deletions.
2 changes: 0 additions & 2 deletions superduper/misc/plugins.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,6 @@ def load_plugin(name: str):
"""
if name == 'local':
return importlib.import_module('superduper.backends.local')
if name == 'ray':
return importlib.import_module('superduper_services.compute.ray.compute')
logging.info(f"Loading plugin: {name}")
plugin = importlib.import_module(f'superduper_{name}')
return plugin
17 changes: 9 additions & 8 deletions test/unittest/component/test_component.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,14 @@ class MyComponent(Component):
a: t.Callable


class UpstreamComponent(Component):
triggered_schedule_jobs = False

def schedule_jobs(self, *args, **kwargs):
self.triggered_schedule_jobs = True
return ('my_dependency_listener',)


def test_init(monkeypatch):
from unittest.mock import MagicMock

Expand Down Expand Up @@ -122,14 +130,7 @@ def test_set_variables(db):


def test_upstream(db):
class MyComponent1(Component):
triggered_schedule_jobs = False

def schedule_jobs(self, *args, **kwargs):
self.triggered_schedule_jobs = True
return ('my_dependency_listener',)

c1 = MyComponent1(identifier='c1')
c1 = UpstreamComponent(identifier='c1')

m = Listener(
identifier='l1',
Expand Down

0 comments on commit c04795c

Please sign in to comment.