Do not import from pyre_extensions (#2937)

Yong Hoon Shin · facebook-github-bot · commit 7e17a6e96f4b · 2024-08-05T18:29:33.000-07:00
Summary: X-link: facebookresearch/FBGEMM#40 Pull Request resolved: #2937 pyre_extensions not availabe in trec OSS. breaks torchrec build tests: https://github.com/pytorch/torchrec/actions/runs/10222011214/job/28285796153 pyre_extensions is also not safe to use in code that will be torch.packaged, as it requires sys for version checks Reviewed By: duduyi2013 Differential Revision: D60792133 fbshipit-source-id: 10ff5f3f4f6a030b0ed82878275a771570595738
diff --git a/fbgemm_gpu/fbgemm_gpu/tbe/ssd/training.py b/fbgemm_gpu/fbgemm_gpu/tbe/ssd/training.py
@@ -36,7 +36,6 @@
 )
 
 from torch import distributed as dist, nn, Tensor  # usort:skip
-from pyre_extensions import none_throws
 from torch.autograd.profiler import record_function
 
 from .common import ASSOC
@@ -450,7 +449,7 @@ def __init__(
         )
         logging.info(
             f"logging stats reporter setup, {self.gather_ssd_cache_stats=}, "
-            f"stats_reporter:{none_throws(self.stats_reporter) if self.stats_reporter else 'none'}, "
+            f"stats_reporter:{self.stats_reporter if self.stats_reporter else 'none'}, "
         )
 
     # pyre-ignore[3]

Original file line number	Diff line number	Diff line change
`@@ -36,7 +36,6 @@`
`36`	`36`	`)`
`37`	`37`
`38`	`38`	`from torch import distributed as dist, nn, Tensor # usort:skip`
`39`		`-from pyre_extensions import none_throws`
`40`	`39`	`from torch.autograd.profiler import record_function`
`41`	`40`
`42`	`41`	`from .common import ASSOC`
`@@ -450,7 +449,7 @@ def __init__(`
`450`	`449`	`)`
`451`	`450`	`logging.info(`
`452`	`451`	`f"logging stats reporter setup, {self.gather_ssd_cache_stats=}, "`
`453`		`- f"stats_reporter:{none_throws(self.stats_reporter) if self.stats_reporter else 'none'}, "`
	`452`	`+ f"stats_reporter:{self.stats_reporter if self.stats_reporter else 'none'}, "`
`454`	`453`	`)`
`455`	`454`
`456`	`455`	`# pyre-ignore[3]`