fix: main() - single START, single END, no variants

inference.py

@@ -405,26 +405,22 @@ def train_agent(num_episodes: int = 10) -> AgentState:
 def main():
     """Main inference loop with RL training."""
     try:
-        # Setup client and get environment variables
-        client, MODEL_NAME = setup_openai_client()
+        print("START")
         
-        # Optional Docker support
+        LOCAL_IMAGE_NAME = os.getenv("LOCAL_IMAGE_NAME")
         if LOCAL_IMAGE_NAME:
-            print(f"START: inference (Docker image: {LOCAL_IMAGE_NAME})")
+            print(f"STEP: inference using Docker image {LOCAL_IMAGE_NAME}")
         else:
-            print("START")
             print("STEP: Starting inference")
         
-        # Run RL training
         agent_state = train_agent(num_episodes=10)
-        print("END: done")
-        logger.info("END: RL training completed successfully")
         return 0
         
     except Exception as e:
-        print("END")
         logger.error(f"Inference failed: {e}")
         return 1
+    finally:
+        print("END")
 
 if __name__ == "__main__":
     exit(main())