Upload folder using huggingface_hub

models.py

@@ -245,21 +245,22 @@ class PythonReviewAction(Action):
 class PythonReviewObservation(Observation):
     """Observation returned by reset/step, including trainer-visible metrics."""
 
-    snippet_id: str
-    code: str
-    filename: str
+    snippet_id: str = ""
+    code: str = ""
+    filename: str = ""
     language: str = "python"
     context: Optional[str] = None
     diff: Optional[str] = None
-    line_count: int = Field(..., ge=0)
-    current_step: int = Field(..., ge=0)
-    max_steps: int = Field(..., ge=1)
-    task_id: str
+    line_count: int = Field(default=0, ge=0)
+    current_step: int = Field(default=0, ge=0)
+    max_steps: int = Field(default=1, ge=1)
+    task_id: str = ""
     review_history: List[ReviewComment] = Field(default_factory=list)
     lines: List[str] = Field(default_factory=list)
     reward_summary: RewardSummary = Field(default_factory=RewardSummary)
     metrics: EpisodeMetrics = Field(default_factory=EpisodeMetrics)
     feedback: str = ""
+    review_time_ms: float = 0.0
     
     # Template compatibility
     task: Optional[TaskDescriptor] = None