[object Object]

← back to Exo

Worker tests 2

7b3e3fd66c1187aa576580242e4d76e74977a0fe · 2025-11-21 16:42:52 +0000 · rltakashige

Files touched

Diff

commit 7b3e3fd66c1187aa576580242e4d76e74977a0fe
Author: rltakashige <rl.takashige@gmail.com>
Date:   Fri Nov 21 16:42:52 2025 +0000

    Worker tests 2
---
 src/exo/worker/engines/mlx/generator/__init__.py                      | 0
 src/exo/worker/{runner => engines/mlx/generator}/generate.py          | 0
 src/exo/worker/runner/runner.py                                       | 2 +-
 src/exo/worker/tests/unittests/__init__.py                            | 0
 src/exo/worker/tests/unittests/{test_plan => }/conftest.py            | 0
 src/exo/worker/tests/unittests/test_download/__init__.py              | 0
 src/exo/worker/tests/unittests/test_mlx/__init__.py                   | 0
 src/exo/worker/tests/unittests/test_plan/test_download_and_loading.py | 2 +-
 src/exo/worker/tests/unittests/test_plan/test_runner_lifecycle.py     | 3 +--
 src/exo/worker/tests/unittests/test_plan/test_task_forwarding.py      | 3 +--
 src/exo/worker/tests/unittests/test_plan/test_warmup.py               | 3 +--
 src/exo/worker/tests/unittests/test_runner/__init__.py                | 0
 12 files changed, 5 insertions(+), 8 deletions(-)

diff --git a/src/exo/worker/engines/mlx/generator/__init__.py b/src/exo/worker/engines/mlx/generator/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/src/exo/worker/runner/generate.py b/src/exo/worker/engines/mlx/generator/generate.py
similarity index 100%
rename from src/exo/worker/runner/generate.py
rename to src/exo/worker/engines/mlx/generator/generate.py
diff --git a/src/exo/worker/runner/runner.py b/src/exo/worker/runner/runner.py
index 81b43524..b99654d6 100644
--- a/src/exo/worker/runner/runner.py
+++ b/src/exo/worker/runner/runner.py
@@ -33,12 +33,12 @@ from exo.shared.types.worker.runners import (
     RunnerWarmingUp,
 )
 from exo.utils.channels import ClosedResourceError, MpReceiver, MpSender
+from exo.worker.engines.mlx.generator.generate import mlx_generate, warmup_inference
 from exo.worker.engines.mlx.utils_mlx import (
     initialize_mlx,
     mlx_force_oom,
 )
 from exo.worker.runner.bootstrap import logger
-from exo.worker.runner.generate import mlx_generate, warmup_inference
 
 
 def main(
diff --git a/src/exo/worker/tests/unittests/__init__.py b/src/exo/worker/tests/unittests/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/src/exo/worker/tests/unittests/test_plan/conftest.py b/src/exo/worker/tests/unittests/conftest.py
similarity index 100%
rename from src/exo/worker/tests/unittests/test_plan/conftest.py
rename to src/exo/worker/tests/unittests/conftest.py
diff --git a/src/exo/worker/tests/unittests/test_download/__init__.py b/src/exo/worker/tests/unittests/test_download/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/src/exo/worker/tests/unittests/test_mlx/__init__.py b/src/exo/worker/tests/unittests/test_mlx/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/src/exo/worker/tests/unittests/test_plan/test_download_and_loading.py b/src/exo/worker/tests/unittests/test_plan/test_download_and_loading.py
index 1cb5adba..d64df456 100644
--- a/src/exo/worker/tests/unittests/test_plan/test_download_and_loading.py
+++ b/src/exo/worker/tests/unittests/test_plan/test_download_and_loading.py
@@ -15,7 +15,7 @@ from exo.worker.tests.constants import (
     RUNNER_1_ID,
     RUNNER_2_ID,
 )
-from exo.worker.tests.unittests.test_plan.conftest import (
+from exo.worker.tests.unittests.conftest import (
     FakeRunnerSupervisor,
     get_mlx_ring_instance,
     get_pipeline_shard_metadata,
diff --git a/src/exo/worker/tests/unittests/test_plan/test_runner_lifecycle.py b/src/exo/worker/tests/unittests/test_plan/test_runner_lifecycle.py
index ce70313c..056de505 100644
--- a/src/exo/worker/tests/unittests/test_plan/test_runner_lifecycle.py
+++ b/src/exo/worker/tests/unittests/test_plan/test_runner_lifecycle.py
@@ -17,8 +17,7 @@ from exo.worker.tests.constants import (
     RUNNER_1_ID,
     RUNNER_2_ID,
 )
-
-from .conftest import (
+from exo.worker.tests.unittests.conftest import (
     FakeRunnerSupervisor,
     get_mlx_ring_instance,
     get_pipeline_shard_metadata,
diff --git a/src/exo/worker/tests/unittests/test_plan/test_task_forwarding.py b/src/exo/worker/tests/unittests/test_plan/test_task_forwarding.py
index 265dd56e..b1500e74 100644
--- a/src/exo/worker/tests/unittests/test_plan/test_task_forwarding.py
+++ b/src/exo/worker/tests/unittests/test_plan/test_task_forwarding.py
@@ -19,8 +19,7 @@ from exo.worker.tests.constants import (
     RUNNER_2_ID,
     TASK_1_ID,
 )
-
-from .conftest import (
+from exo.worker.tests.unittests.conftest import (
     FakeRunnerSupervisor,
     OtherTask,
     get_mlx_ring_instance,
diff --git a/src/exo/worker/tests/unittests/test_plan/test_warmup.py b/src/exo/worker/tests/unittests/test_plan/test_warmup.py
index 601f4987..ed0f0d2b 100644
--- a/src/exo/worker/tests/unittests/test_plan/test_warmup.py
+++ b/src/exo/worker/tests/unittests/test_plan/test_warmup.py
@@ -14,8 +14,7 @@ from exo.worker.tests.constants import (
     RUNNER_1_ID,
     RUNNER_2_ID,
 )
-
-from .conftest import (
+from exo.worker.tests.unittests.conftest import (
     FakeRunnerSupervisor,
     get_mlx_ring_instance,
     get_pipeline_shard_metadata,
diff --git a/src/exo/worker/tests/unittests/test_runner/__init__.py b/src/exo/worker/tests/unittests/test_runner/__init__.py
new file mode 100644
index 00000000..e69de29b

← de508113 Worker tests on staging 1  ·  back to Exo  ·  bump pyo3 stub-gen 7088988a →