diff --git a/docs/source/using-the-python-api.mdx b/docs/source/using-the-python-api.mdx index 797f21740..81fb3e4d9 100644 --- a/docs/source/using-the-python-api.mdx +++ b/docs/source/using-the-python-api.mdx @@ -31,7 +31,7 @@ def main(): pipeline_params = PipelineParameters( launcher_type=ParallelismManager.ACCELERATE, - custom_task_directory=None, # if using a custom task + custom_tasks_directory=None, # if using a custom task # Remove the 2 parameters below once your configuration is tested max_samples=10 )