Upload folder using huggingface_hub

Inference.py

@@ -209,9 +209,8 @@ async def run_episode_loop(
         actions = await call_llm_for_actions(client, user_prompt)
         action_payload = actions_to_hft_action(actions)
         observation = await step_fn(action_payload)
-        observation, user_prompt = build_user_prompt(
-            observation.get("observation", observation)
-        )
+        log_transcript(f"Received observation: {observation}")
+        observation, user_prompt = build_user_prompt(observation.get("observation"))
 
         reward = observation.reward
         done = observation.done

client.py

@@ -80,9 +80,8 @@ class HftEnv(EnvClient[HftAction, HftObservation, HftState]):
         obs_data = payload.get("observation", {})
         observation = HftObservation(
             time=obs_data.get("time", 0.0),
-            reward=obs_data.get("reward", 0.0),
+            reward=payload.get("reward", 0.0),
             done=obs_data.get("done", False),
-            spread=obs_data.get("spread"),
             history=obs_data.get("history", []),
             active_orders=obs_data.get("active_orders", []),
         )