Final big migration push [3/22]

Differential Revision: D54731104
connernilsen · Mar 11, 2024 · 9b27559 · 9b27559
1 parent bd96e8f
commit 9b27559
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/torchrec/distributed/train_pipeline/train_pipeline.py b/torchrec/distributed/train_pipeline/train_pipeline.py
@@ -7,6 +7,11 @@
 
 # pyre-strict
 
+"""
+NOTE: Due to an internal packaging issue, `train_pipeline.py` must be compatible with
+older versions of TorchRec. Importing new modules from other files may break model
+publishing flows.
+"""
 import abc
 import logging
 from typing import cast, Generic, Iterator, List, Optional, Tuple