extra_script is missing in SLURM(BaseScheduler) about adaptive-scheduler HOT 6 CLOSED

scheduler_kwargs = dict(
        num_threads = 4,
        extra_scheduler=[
            "--exclusive", 
            "--partition=partition"
        ],
        extra_script="umask 0000", # or ["umask 0000"]
        executor_type="process-pool",
    )
import adaptive_scheduler
sch = adaptive_scheduler.scheduler.SLURM(cores=1, **scheduler_kwargs)
print(sch.job_script())

#!/bin/bash
#SBATCH --ntasks 1
#SBATCH --no-requeue
#SBATCH --exclusive
#SBATCH --partition=partition

export MKL_NUM_THREADS=4
export OPENBLAS_NUM_THREADS=4
export OMP_NUM_THREADS=4
export NUMEXPR_NUM_THREADS=4


umask 0000

/Users/basnijholt/miniconda3/bin/python run_learner.py --n 1 --log-fname ${NAME}-${SLURM_JOB_ID}.log --job-id ${SLURM_JOB_ID} --name ${NAME}

import concurrent.futures
ex = concurrent.futures.ProcessPoolExecutor()
# Ignores extra_script
run_manager.ioloop.run_in_executor(ex, scheduler.write_job_script, "test")
# Correctly uses extra_script
scheduler.write_job_script("test2")

import pickle
print(scheduler.extra_script)  # prints correctly
print(pickle.loads(pickle.dumps(scheduler)).extra_script)  # prints empty string