Upload folder using huggingface_hub

client.py

@@ -53,10 +53,18 @@ class NL2SQLEnv:
     def _parse_result(self, payload: Dict[str, Any]) -> StepResult:
         obs_data = payload.get("observation", {})
         
-        # Read standard OpenEnv top-level keys safely
-        safe_reward = float(payload.get("reward", 0.0))
-        safe_done = bool(payload.get("done", False))
+        # SAFETY CHECK: Handle JSON 'null' (None) values gracefully
+        raw_reward = payload.get("reward")
+        safe_reward = float(raw_reward) if raw_reward is not None else 0.0
         
+        raw_obs_reward = obs_data.get("reward")
+        safe_obs_reward = float(raw_obs_reward) if raw_obs_reward is not None else 0.0
+        
+        raw_score = obs_data.get("score")
+        safe_score = float(raw_score) if raw_score is not None else 0.0
+        
+        safe_done = bool(payload.get("done") or obs_data.get("done") or False)
+
         obs = NL2SQLObservation(
             question=obs_data.get("question", ""),
             schema_context=obs_data.get("schema_context", ""),
@@ -68,8 +76,8 @@ class NL2SQLEnv:
             step=obs_data.get("step", 0),
             max_steps=obs_data.get("max_steps", 5),
             done=safe_done,
-            reward=safe_reward,
-            score=float(obs_data.get("score", 0.0)),
+            reward=safe_obs_reward,
+            score=safe_score,
         )
         return StepResult(
             observation=obs,