Rename train_pipeline to train_pipelines (#1795)

Summary: Pull Request resolved: #1795 Rename things. Reviewed By: joshuadeng Differential Revision: D54864192 fbshipit-source-id: 340f6b5b375563270dbba823f1ba1a38c268dd66
pytorch · Mar 14, 2024 · 32e0e14 · 32e0e14
1 parent 9d03945
commit 32e0e14
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/torchrec/distributed/train_pipeline/__init__.py b/torchrec/distributed/train_pipeline/__init__.py
@@ -8,7 +8,7 @@
 # pyre-strict
 
 
-from torchrec.distributed.train_pipeline.train_pipeline import (  # noqa
+from torchrec.distributed.train_pipeline.train_pipelines import (  # noqa
     EvalPipelineSparseDist,  # noqa
     PrefetchTrainPipelineSparseDist,  # noqa
     StagedTrainPipeline,  # noqa

diff --git a/torchrec/distributed/train_pipeline/tests/test_train_pipeline.py b/torchrec/distributed/train_pipeline/tests/test_train_pipeline.py
@@ -45,7 +45,7 @@
 from torchrec.distributed.tests.test_fp_embeddingbag_utils import (
     create_module_and_freeze,
 )
-from torchrec.distributed.train_pipeline.train_pipeline import (
+from torchrec.distributed.train_pipeline.train_pipelines import (
     EvalPipelineSparseDist,
     PrefetchTrainPipelineSparseDist,
     StagedTrainPipeline,

diff --git a/...tributed/train_pipeline/train_pipeline.py → ...ributed/train_pipeline/train_pipelines.py b/...tributed/train_pipeline/train_pipeline.py → ...ributed/train_pipeline/train_pipelines.py