inference issues

inference.py

@@ -39,7 +39,7 @@ API_KEY: str = os.environ["HF_TOKEN"]
 
 TASK_NAME: str = os.getenv("FITSCRIPT_TASK", "basic_plan")
 BENCHMARK: str = "fitscript_env"
-IMAGE_NAME: str = os.getenv("FITSCRIPT_IMAGE", "FitScript-env:latest")
+IMAGE_NAME: str = os.getenv("FITSCRIPT_IMAGE") or os.getenv("LOCAL_IMAGE_NAME", "FitScript-env:latest")
 MAX_STEPS: int = int(os.getenv("MAX_STEPS", "8"))
 
 # ---------------------------------------------------------------------------
@@ -167,7 +167,12 @@ async def run_episode() -> None:
 
     env = None
     try:
-        env = FitscriptEnv.from_docker_image(IMAGE_NAME)
+        USE_DOCKER = os.getenv("USE_DOCKER", "true").lower() == "true"
+
+        if USE_DOCKER:
+            env = await FitscriptEnv.from_docker_image(IMAGE_NAME)
+        else:
+            env = FitscriptEnv(base_url="http://localhost:8000")
 
         # Reset
         reset_result = env.reset()
@@ -234,7 +239,10 @@ async def run_episode() -> None:
         print(f"[ERROR] {error_msg}", flush=True, file=sys.stderr)
     finally:
         if env is not None:
-            env.close()
+            if USE_DOCKER:
+                await env.close()
+            else:
+                env.close()
 
     log_end(success, step, final_score, rewards)
 

openenv_FitScript.egg-info/PKG-INFO

@@ -0,0 +1,9 @@
+Metadata-Version: 2.4
+Name: openenv-FitScript
+Version: 0.1.0
+Summary: Fitscript environment for OpenEnv
+Requires-Python: >=3.10
+Requires-Dist: openenv-core[core]>=0.2.2
+Provides-Extra: dev
+Requires-Dist: pytest>=8.0.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0.0; extra == "dev"

openenv_FitScript.egg-info/SOURCES.txt

@@ -0,0 +1,15 @@
+README.md
+pyproject.toml
+./__init__.py
+./client.py
+./inference.py
+./models.py
+openenv_FitScript.egg-info/PKG-INFO
+openenv_FitScript.egg-info/SOURCES.txt
+openenv_FitScript.egg-info/dependency_links.txt
+openenv_FitScript.egg-info/entry_points.txt
+openenv_FitScript.egg-info/requires.txt
+openenv_FitScript.egg-info/top_level.txt
+server/FitScript_environment.py
+server/__init__.py
+server/app.py

openenv_FitScript.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

openenv_FitScript.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+server = FitScript.server.app:main

openenv_FitScript.egg-info/requires.txt

@@ -0,0 +1,5 @@
+openenv-core[core]>=0.2.2
+
+[dev]
+pytest>=8.0.0
+pytest-cov>=4.0.0

openenv_FitScript.egg-info/top_level.txt

@@ -0,0 +1 @@
+FitScript