Slurm agent
Installation
To install the Slurm agent, run the following command:
$ pip install flytekitplugins-slurmExample usage
For the example usage of different Slurm task types, please see Slurm agent example usage.
Local testing
To test the Slurm agent locally, create a class for the agent task that inherits from AsyncAgentExecutorMixin. This mixin can handle asynchronous tasks and allows flytekit to mimic FlytePropeller’s behavior in calling the agent. For more information, see “Testing agents locally”.
In some cases, you will need to store credentials in your local environment when testing locally.
Flyte deployment configuration
If you are using a managed deployment of Flyte, you will need to contact your deployment administrator to configure agents in your deployment.
To enable the Slurm agent in your Flyte deployment, see the Slurm agent deployment guide.