#!/bin/bash
#SBATCH --job-name=probe10_lectin
#SBATCH --partition=nova
#SBATCH --gres=gpu:a100:1
#SBATCH --cpus-per-task=8
#SBATCH --mem=64G
#SBATCH --time=02:00:00
#SBATCH --output=bert_v6_contrastive/analysis/probing_analysis/slurm_logs/probe10_%j.out
#SBATCH --error=bert_v6_contrastive/analysis/probing_analysis/slurm_logs/probe10_%j.err

echo "========================================"
echo "Probe 10: Lectin Motif Recognition"
echo "Job ID: $SLURM_JOB_ID"
echo "Node:   $(hostname)"
echo "Start:  $(date)"
echo "========================================"

# Activate environment
source /work/ratul1/supantha/miniconda3/etc/profile.d/conda.sh
conda activate glycanml

cd /work/ratul1/supantha/glycan-SD-VS/bert_training_v3/v3.1_cluster_training

echo "Python: $(which python3)"
echo "PyTorch: $(python3 -c 'import torch; print(torch.__version__, "CUDA:", torch.cuda.is_available())')"
nvidia-smi --query-gpu=name,memory.total --format=csv,noheader 2>/dev/null || echo "No GPU"
echo ""

# Ensure output dirs exist
mkdir -p bert_v6_contrastive/analysis/probing_analysis/slurm_logs
mkdir -p bert_v6_contrastive/analysis/probing_analysis/10_lectin_motifs_v5
mkdir -p bert_v6_contrastive/analysis/probing_analysis/10_lectin_motifs_v6

# Install deps if needed
pip install -q scikit-learn 2>/dev/null

# Run probe 10 (runs both V5-A and V6 internally)
python3 -u bert_v6_contrastive/scripts/probe_10_lectin_motifs.py \
    --device cuda

echo ""
echo "Completed: $(date)"
echo "Output dirs:"
ls -la bert_v6_contrastive/analysis/probing_analysis/10_lectin_motifs_v5/ 2>/dev/null
ls -la bert_v6_contrastive/analysis/probing_analysis/10_lectin_motifs_v6/ 2>/dev/null