#!/bin/bash

# Exit immediately if a command exits with a non-zero status
set -e
export CUDA_VISIBLE_DEVICES=7
# Path to the python executable in the conda environment
PYTHON_EXEC="/home/debarpanb1/anaconda3/envs/irl_torch2/bin/python"
SCRIPT_PATH="/home/debarpanb1/TREA_2.0/pipeline/generate_reasoning_traces.py"
DATASET_DIR="/home/debarpanb1/TREA_2.0/pipeline/dataset_v5"
TRACE_TEMPLATES="/home/debarpanb1/TREA_2.0/pipeline/trace_templates.json"
CONFIG_PATH="/home/debarpanb1/TREA_2.0/pipeline/config.yaml"

echo "================================================================"
echo "Starting generation of multihop reasoning traces using Llama 3.1"
echo "Environment: irl_torch2"
echo "Dataset: $DATASET_DIR"
echo "================================================================"

# Run the python script, passing any additional arguments pro   vided to this shell script
# E.g., ./run_multihop_reasoning_traces.sh --dry_run --tasks multi_hop
$PYTHON_EXEC $SCRIPT_PATH \
    --dataset_dir "$DATASET_DIR" \
    --trace_templates "$TRACE_TEMPLATES" \
    --config "$CONFIG_PATH" \
    --tasks duration_gap temporal_arithmetic temporal_loudness multi_hop

echo "================================================================"
echo "Completed generation of multihop reasoning traces."
echo "================================================================"