refactor: improve logging level for episode registration exit warning

Author: binary-husky

ajet/task_rollout/single_worker.py

@@ -1,6 +1,7 @@
 """Single worker primitives for environment rollouts."""
 
 import uuid
+import time
 import threading
 from typing import Literal
 
@@ -168,7 +169,6 @@ def rollout_env_worker_loop(
             cnt = 1
 
             while True:
-
                 if observation_window["stop"][task_thread_index]:           # since we use multi-threading, the best way to communicate with main thread is through shared memory.
                     observation_window["info"][task_thread_index] += f"[thread {task_thread_index} observe stop, returning]\n"
                     return
@@ -205,10 +205,9 @@ def rollout_env_worker_loop(
                     observation_window["info"][task_thread_index] += f"[thread {task_thread_index} observe stop, returning]\n"
                     return
                 else:
+                    time.sleep(0)   # be nice to other threads
                     continue
 
         except Exception as e:
-            logger.exception(
-                f"encounter exception in env_worker_loop error={e.args}"
-            )
+            logger.exception(f"encounter exception in env_worker_loop error={e.args}")
             raise e

ajet/tuner_lib/experimental/interchange_utils.py

@@ -158,7 +158,7 @@ def http_register_episode(config,
                           should_exit_soft):
 
     if should_exit_soft():
-        logger.warning(f"Exiting before registering episode {episode_uuid}")
+        logger.debug(f"Exiting before registering episode {episode_uuid}")
         return None
 
     # parse episode_uuid, openai_base_url, openai_api_key