2025-05-31 nightly release (151aa02)

pytorchbot · pytorchbot · commit 8b2d00f707be · 2025-05-31T11:34:56.000Z
diff --git a/torchrec/modules/itep_modules.py b/torchrec/modules/itep_modules.py
@@ -108,7 +108,7 @@ def __init__(
             self.last_pruned_iter: int = -1
             self.pg: Optional[dist.ProcessGroup] = pg
 
-            if self.lookups is not None:
+            if self.lookups:
                 self.init_itep_state()
             else:
                 logger.info(
