[ci-fix] Fix failing doc-build due to sagemaker scheduler from #801

pytorch · Mar 27, 2024 · 495bb00 · 495bb00
1 parent 2dbfec1
commit 495bb00
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchx/schedulers/aws_sagemaker_scheduler.py b/torchx/schedulers/aws_sagemaker_scheduler.py
@@ -167,7 +167,7 @@ class AWSSageMakerScheduler(DockerWorkspaceMixin, Scheduler[AWSSageMakerOpts]):
     **Config Options**
 
     .. runopts::
-        class: torchx.schedulers.sagemaker_scheduler.create_scheduler
+        class: torchx.schedulers.aws_sagemaker_scheduler.create_create_scheduler
 
     **Compatibility**