wip

tests/test_training.py

@@ -11,6 +11,7 @@ import sys
 import asyncio
 from pathlib import Path
 from datetime import datetime, timedelta
+from safe_logging import setup_safe_logging
 
 # Add project root to path
 project_root = Path(__file__).parent