Remove dynamo runner's dependency on distributed build (#117903)

Summary:
So that we can bisect faster without needing to rebuild distributed module. We remove the annotation to avoid flake8 undefined name lint

X-link: pytorch/pytorch#117903
Approved by: https://github.com/xuzhao9

Reviewed By: DanilBaibak

Differential Revision: D53082802

Pulled By: xmfan

fbshipit-source-id: 9c3421ee69d76925c2995b985a9bfee632242f1b

userbenchmark/dynamo/dynamobench/common.py

@@ -60,7 +60,6 @@
     reset_rng_state,
     same,
 )
-from torch.distributed.fsdp.wrap import ModuleWrapPolicy
 
 try:
     from torch._dynamo.utils import (
@@ -2195,7 +2194,7 @@ def get_benchmark_indices(self, length):
         )
         return start, end
 
-    def get_fsdp_auto_wrap_policy(self, model_name: str) -> Optional[ModuleWrapPolicy]:
+    def get_fsdp_auto_wrap_policy(self, model_name: str):
         from diffusers.models.transformer_2d import Transformer2DModel
 
         from torch.distributed.fsdp.wrap import (