Run attacks continuously

inference.py

@@ -1,5 +1,6 @@
 import threading
 import time
+import sys
 
 from env.fake_server import run_server
 from tasks.easy import run as easy
@@ -13,35 +14,36 @@ server_thread.start()
 
 time.sleep(2)  # wait for server to start
 
+print("[START] task=ai-deception env=cyber-security model=baseline", flush=True)
 
-print("[START] task=ai-deception env=cyber-security model=baseline")
-
-
-# Run continuously for Hugging Face
+# Run continuously
 while True:
 
     # Easy Task
     easy_score = easy()
     print(
-        f"[STEP] step=1 action=easy reward={easy_score:.2f} done=false error=null"
+        f"[STEP] step=1 action=easy reward={easy_score:.2f} done=false error=null",
+        flush=True
     )
 
     # Medium Task
     medium_score = medium()
     print(
-        f"[STEP] step=2 action=medium reward={medium_score:.2f} done=false error=null"
+        f"[STEP] step=2 action=medium reward={medium_score:.2f} done=false error=null",
+        flush=True
     )
 
     # Hard Task
     hard_score = hard()
     print(
-        f"[STEP] step=3 action=hard reward={hard_score:.2f} done=true error=null"
+        f"[STEP] step=3 action=hard reward={hard_score:.2f} done=true error=null",
+        flush=True
     )
 
     print(
-        f"[END] success=true steps=3 rewards={easy_score:.2f},{medium_score:.2f},{hard_score:.2f}"
+        f"[END] success=true steps=3 rewards={easy_score:.2f},{medium_score:.2f},{hard_score:.2f}",
+        flush=True
     )
 
-    # Wait before next attack cycle
+    # wait before next attack cycle
     time.sleep(60)
-