#!/bin/bash
# 快速测试脚本

echo "=========================================="
echo "测试 MAGAIL 训练脚本"
echo "=========================================="

# 激活conda环境并运行
cd /home/huangfukk/MAGAIL4AutoDrive

echo ""
echo "1. 测试模块导入..."
python -c "from Algorithm.magail import MAGAIL; print('✅ 导入成功!')" || exit 1

echo ""
echo "2. 运行训练（1个episode，不渲染）..."
python train_magail.py --episodes 1 --horizon 50 2>&1 | tail -30

echo ""
echo "=========================================="
echo "测试完成！"
echo "=========================================="