Fix tests.

Author: np-complete0

.github/workflows/ci-python.yml

@@ -63,10 +63,10 @@ jobs:
           cd ${{ github.workspace }}/apps
           uv run pyright
       
-      - name: Run tests (pytest) for bfd-pipeline-idr
+      - name: Run tests (pytest)
         run: |
           cd ${{ github.workspace }}/apps
-          uv run pytest bfd-pipeline/bfd-pipeline-idr
+          uv run pytest
   
   non-workspace-checks:
     name: Non-Workspace Project - sftp-outbound-transfer

apps/bfd-pipeline/bfd-pipeline-idr/pipeline.py

@@ -2,7 +2,6 @@
 import os
 import sys
 import time
-from pathlib import Path
 
 import pipeline_nodes
 import ray
@@ -28,20 +27,26 @@ def main() -> None:
     logger.info("load start")
 
     parallelism = int(os.environ.get("PARALLELISM", "6"))
-    # Shutdown any existing Ray instance first to avoid conflicts.
-    if ray.is_initialized():  # type: ignore
-        ray.shutdown()  # type: ignore
-        time.sleep(1)
-    ray.init(
-        logging_level="info",
-        num_cpus=parallelism,
-        local_mode=True,  # Run tasks inprocess to avoid worker version mismatches
-        ignore_reinit_error=True,
-        runtime_env=None,
-    )  # type: ignore
-
-    dict_builder = base.DictResult()
-    adapter = RayGraphAdapter(result_builder=dict_builder)
+    mode = sys.argv[1] if len(sys.argv) > 1 else ""
+    use_ray = mode not in ("synthetic", "local")
+
+    if use_ray:
+        # Shutdown any existing Ray instance first to avoid conflicts.
+        if ray.is_initialized():
+            ray.shutdown()
+            time.sleep(1)
+
+        ray.init(
+            logging_level="info",
+            num_cpus=parallelism,
+        )
+        dict_builder = base.DictResult()
+        adapter = RayGraphAdapter(result_builder=dict_builder)
+    else:
+        # Use simple non distributed adapter for tests
+        dict_builder = base.DictResult()
+        adapter = base.SimplePythonGraphAdapter(result_builder=dict_builder)
+
     load_type = str(os.environ.get("LOAD_TYPE", "incremental"))
     logger.info("load_type %s", load_type)
     dr = (

apps/pyproject.toml

@@ -90,5 +90,4 @@ reportMissingTypeStubs = false
 
 [tool.pytest.ini_options]
 timeout = 600
-testpaths = ["bfd-pipeline/bfd-pipeline-idr"]