78 lines
2.7 KiB
Python
78 lines
2.7 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Simple Enhanced Training Test
|
|
|
|
Quick test to verify enhanced training system can be enabled and controlled.
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
import logging
|
|
|
|
# Add project root to path
|
|
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
from core.orchestrator import TradingOrchestrator
|
|
from core.data_provider import DataProvider
|
|
|
|
# Configure logging
|
|
logging.basicConfig(level=logging.INFO)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
def test_enhanced_training():
|
|
"""Test enhanced training system"""
|
|
try:
|
|
logger.info("Testing Enhanced Training System...")
|
|
|
|
# 1. Create data provider
|
|
data_provider = DataProvider()
|
|
|
|
# 2. Create orchestrator with enhanced training ENABLED
|
|
logger.info("Creating orchestrator with enhanced_rl_training=True...")
|
|
orchestrator = TradingOrchestrator(
|
|
data_provider=data_provider,
|
|
enhanced_rl_training=True # 🔥 THIS ENABLES IT
|
|
)
|
|
|
|
# 3. Check if training system is available
|
|
logger.info(f"Training system available: {orchestrator.enhanced_training_system is not None}")
|
|
logger.info(f"Training enabled: {orchestrator.training_enabled}")
|
|
|
|
# 4. Get training stats
|
|
stats = orchestrator.get_enhanced_training_stats()
|
|
logger.info(f"Training stats: {stats}")
|
|
|
|
# 5. Test start/stop
|
|
if orchestrator.enhanced_training_system:
|
|
logger.info("Testing start/stop functionality...")
|
|
|
|
# Start training
|
|
start_result = orchestrator.start_enhanced_training()
|
|
logger.info(f"Start result: {start_result}")
|
|
|
|
# Get updated stats
|
|
updated_stats = orchestrator.get_enhanced_training_stats()
|
|
logger.info(f"Updated stats: {updated_stats}")
|
|
|
|
# Stop training
|
|
stop_result = orchestrator.stop_enhanced_training()
|
|
logger.info(f"Stop result: {stop_result}")
|
|
|
|
logger.info("✅ Enhanced training system is working!")
|
|
return True
|
|
else:
|
|
logger.warning("❌ Enhanced training system not available")
|
|
return False
|
|
|
|
except Exception as e:
|
|
logger.error(f"Error testing enhanced training: {e}")
|
|
return False
|
|
|
|
if __name__ == "__main__":
|
|
success = test_enhanced_training()
|
|
if success:
|
|
print("\n🎉 Enhanced training system is ready to use!")
|
|
print("To enable it in your main system, use:")
|
|
print(" enhanced_rl_training=True when creating TradingOrchestrator")
|
|
else:
|
|
print("\n⚠️ Enhanced training system has issues. Check the logs above.") |