[shardformer] Run meaningful doc test

hpcaitech · Dec 25, 2024 · 52a3b88 · 52a3b88
1 parent c4df1cc
commit 52a3b88
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/source/en/features/zerobubble_pipeline_parallelism.md b/docs/source/en/features/zerobubble_pipeline_parallelism.md
@@ -235,4 +235,4 @@ Performance Benchmark
 ## API Reference
 {{ autodoc:colossalai.pipeline.schedule.zero_bubble_pp.ZeroBubbleVPipeScheduler }}
 
-<!-- doc-test-command: echo  -->
+<!-- doc-test-command: python -m pytest -s tests/test_pipeline/test_schedule/test_zerobubble_pp.py  -->
diff --git a/docs/source/zh-Hans/features/zerobubble_pipeline_parallelism.md b/docs/source/zh-Hans/features/zerobubble_pipeline_parallelism.md
@@ -234,4 +234,4 @@ plugin = HybridParallelPlugin(
 ## API 参考
 {{ autodoc:colossalai.pipeline.schedule.zero_bubble_pp.ZeroBubbleVPipeScheduler }}
 
-<!-- doc-test-command: echo  -->
+<!-- doc-test-command: python -m pytest -s tests/test_pipeline/test_schedule/test_zerobubble_pp.py  -->