From 9b275599a3824aa3cbe2e17489db31826607f4d1 Mon Sep 17 00:00:00 2001 From: Conner Nilsen Date: Mon, 11 Mar 2024 10:14:17 -0700 Subject: [PATCH] Final big migration push [3/22] Differential Revision: D54731104 --- torchrec/distributed/train_pipeline/train_pipeline.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/torchrec/distributed/train_pipeline/train_pipeline.py b/torchrec/distributed/train_pipeline/train_pipeline.py index 657488baf..ae543a96f 100644 --- 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