/festive-pelican

A lil python package to make dask easier on the W&M HPC

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

festive-pelican

A lil python package to make dask easier on the W&M HPC

How to use

This package currently provides a function and a class that take the same arguments: hpc_dask_cluster and HPCDaskTaskRunner. The former returns a dask_jobqueue PBSCluster, and the latter inherits Prefect's DaskTaskRunner.

Here are the arguments they take:

num_procs: int,                        # number of processes you'd like
job_name: str,                         # name of PBS job
cluster: str,                          # name of cluster, right now this can just be "vortex"
cores_per_process: Optional[int]=None, # automatically scale 
walltime: str = "01:00:00",            # walltime of PBS job
**kwargs                               # anything else you'd like to pass to PBSCluster

License

This project is licensed under GPLv3 or any later version. See LICENSE.md for more information.