#!/usr/bin/zsh

### Add basic configuration for job

#SBATCH --job-name=dataset_generation
#SBATCH --output=logs/dataset_generation_%j.log
#SBATCH --error=logs/dataset_generation_error_%j.log
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=6
#SBATCH --gres=gpu:1
#SBATCH --time=20:00:00


###------------------------------------------------------------------------------------------------------------------------------

### Run the project in work directory of the cluster (configure based on need!! 
### RWTH File System : https://help.itc.rwth-aachen.de/en/service/rhr4fjjutttf/article/da307ec2c60940b29bd42ac483fc3ea7/
cd $HPCWORK
cd codebud/data_generation
###------------------------------------------------------------------------------------------------------------------------------

### JOB SCRIPT RUN
module load GCCcore/.13.2.0
module load Python/3.11.5
module load CUDA

source ../../venvs/codebud/bin/activate
echo $VIRTUAL_ENV

python --version
python main.py

module unload CUDA
module unload Python/3.11.5

deactivate
echo "Script ran successfully"