fixed a minor bug in setting the expression in the scheduler

AynurAda · Sep 16, 2024 · 4148f48 · 4148f48
1 parent 1be109c
commit 4148f48
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,3 @@
+symbolicai
+pytest
+rpyc
diff --git a/src/__init__.py b/src/__init__.py
diff --git a/src/func.py b/src/func.py
@@ -38,7 +38,7 @@ def __init__(self, expr: Expression, num_workers: int, engine: Callable, dataset
         self.llm_responses: Dict[int, Optional[Any]] = {}
         self.llm_response_ready: Dict[int, threading.Event] = {}
         self.pending_tasks: int = len(self.dataset)
-        self.expr: Expression = expr
+        self.expr: Expression = expr()
         self.pending_tasks_update: threading.Event = threading.Event()
 
     def single_expression(self, data_point: Any) -> Any:

diff --git a/tests/tests_batchscheduler.py → tests/test_batchscheduler.py b/tests/tests_batchscheduler.py → tests/test_batchscheduler.py
@@ -5,9 +5,12 @@
 
 from symai import Expression, Symbol
 from symai.backend.base import BatchEngine
-from src.func import BatchScheduler
 from symai.functional import EngineRepository   
 
+import sys
+import os
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
+from src.func import BatchScheduler
 
 class TestExpression(Expression):
     def __init__(self, **kwargs):