Upload folder using huggingface_hub

server/environment.py

@@ -152,7 +152,7 @@ class NL2SQLEnvironment(Environment):
         self._last_obs = obs
         return obs
 
-    def step(self, action: NL2SQLAction) -> tuple[NL2SQLObservation, float, bool, dict]:
+    def step(self, action: NL2SQLAction) -> NL2SQLObservation:
         """Execute the agent's SQL and return graded observation."""
         if self._task is None or self._example is None:
             # Called before reset — auto-reset
@@ -212,9 +212,9 @@ class NL2SQLEnvironment(Environment):
         )
         self._last_obs = obs
         
-        # CRITICAL FIX: Return the 4-part OpenEnv tuple!
-        info = {"score": score, "error": error}
-        return obs, reward, done, info
+        # openenv-core expects ONLY the observation returned from step().
+        # The framework reads obs.reward and obs.done itself — do NOT return a tuple.
+        return obs
 
     @property
     def state(self) -> NL2SQLState:
@@ -223,4 +223,4 @@ class NL2SQLEnvironment(Environment):
     # ── Helpers ────────────────────────────────────────────────────────────
 
     def available_tasks(self) -> list:
-        return all_task_names()
+        return all_task_names()