Merge pull request #790 from expectedparrot/serialization

Add missing Jobs imports
expectedparrot · Jul 22, 2024 · d778781 · d778781
2 parents 2fe050c + 7837d45
commit d778781
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/edsl/jobs/Jobs.py b/edsl/jobs/Jobs.py
@@ -461,6 +461,13 @@ def run(
             remote_inference = False
 
         if remote_inference:
+            from edsl.agents.Agent import Agent
+            from edsl.language_models.registry import Model
+            from edsl.results.Result import Result
+            from edsl.results.Results import Results
+            from edsl.scenarios.Scenario import Scenario
+            from edsl.surveys.Survey import Survey
+
             self._output("Remote inference activated. Sending job to server...")
             if remote_cache:
                 self._output(

diff --git a/tests/serialization/data/0.1.30.json b/tests/serialization/data/0.1.30.json