#!/bin/bash
#SBATCH --job-name=activation
#SBATCH --gres=gpu:1
#SBATCH --cpus-per-task=8
#SBATCH --partition=physical-gpu
#SBATCH --mem=32G
#SBATCH --output=logs/%j-activation.out

ENV_FILE=$1

if [ -z "$ENV_FILE" ]; then
    echo "Error: No environment file specified"
    exit 1
fi

if [ ! -f "$ENV_FILE" ]; then
    echo "Error: Environment file $ENV_FILE not found"
    exit 1
fi

# Load the specified environment file
source "$ENV_FILE"

# Create the artifact folder
mkdir -p "$OUTPUT_DIR"

export PYTHONUNBUFFERED=1

# Run the Python script
python ../activation.py \
    --model $MODEL \
    --output-dir $OUTPUT_DIR \
    --data-path $DATA_PATH \
    --lang $LANG

echo "Activation single job done"