final bugs and removed debug

baseline/run_baseline.py

@@ -319,7 +319,6 @@ def get_model_action(client: Any, observation: Observation, step_count: int) ->
         response = (completion.choices[0].message.content or "").strip()
         return parse_action_from_response(response)
     except Exception as exc:  # pragma: no cover - API/network dependent
-        print(f"[DEBUG] Model request failed: {exc}", flush=True)
         return heuristic_next_action(observation, step_count)
 
 

env/taint_tracker.py

@@ -49,6 +49,7 @@ class TaintTracker:
                 artifact_body=text,
                 destination="internal-db",
                 contamination_classes=[],
+                semantic_labels=[],
                 leak_detected=False,
                 leak_severity="none",
             )