#!/bin/bash
# MAMBA2 Hyperopt RunPod Deployment Script
# Deploys MAMBA2 hyperparameter optimization to RunPod GPU

set -e

# Activate virtual environment
source .venv/bin/activate

# Set PYTHONPATH to include custom runpod module
export PYTHONPATH=/home/jgrusewski/Work/foxhunt:$PYTHONPATH

# Configuration
GPU_TYPE="RTX A4000"
POD_NAME="mamba2-hyperopt"
IMAGE="jgrusewski/foxhunt:latest"
TRIALS=50
EPOCHS=50
TIMEOUT="120m"

# MAMBA2 hyperopt command for RunPod
# Note: Binary is wrapped by entrypoint-self-terminate.sh
COMMAND="hyperopt_mamba2_demo \
  --parquet-file /runpod-volume/test_data/ES_FUT_180d.parquet \
  --base-dir /runpod-volume/ml_training \
  --trials ${TRIALS} \
  --epochs ${EPOCHS} \
  --batch-size-max 96 \
  --early-stopping-patience 5"

echo "======================================================================"
echo "MAMBA2 Hyperopt RunPod Deployment"
echo "======================================================================"
echo "GPU Type: ${GPU_TYPE}"
echo "Docker Image: ${IMAGE}"
echo "Trials: ${TRIALS}"
echo "Epochs per Trial: ${EPOCHS}"
echo "Max Monitoring: ${TIMEOUT}"
echo "Command: ${COMMAND}"
echo "======================================================================"
echo ""

# Deploy pod with monitoring and auto-stop
python3 scripts/runpod_deploy.py \
  --gpu-type "${GPU_TYPE}" \
  --image "${IMAGE}" \
  --command "${COMMAND}" \
  --monitor \
  --auto-stop \
  --timeout "${TIMEOUT}" \
  --monitor-interval 15

echo ""
echo "======================================================================"
echo "Deployment Complete!"
echo "======================================================================"
echo "Results will be saved to: /runpod-volume/ml_training/"
echo "Check S3 bucket for outputs: s3://se3zdnb5o4/ml_training/"
echo "======================================================================"