update DTensor namespace for non public APIs in TorchRec (pytorch#2314)

Summary:
Pull Request resolved: pytorch#2314

Interim fix to use torchrec LocalShardsWrapper to fix OSS CI for `_shards_wrapper` while PT distributed tensor namespace changes sync to internal.

Reviewed By: PaulZhang12

Differential Revision: D61483729

fbshipit-source-id: 42f2fe7e5234bc327966ffbe67c4750606f15595

torchrec/distributed/tests/test_shards_wrapper.py

@@ -11,9 +11,8 @@
 from typing import List, Optional, Union
 
 import torch
-from hypothesis import settings, Verbosity
 from torch import distributed as dist
-from torch.distributed._tensor._shards_wrapper import LocalShardsWrapper
+from torchrec.distributed.shards_wrapper import LocalShardsWrapper
 from torchrec.distributed.test_utils.multi_process import (
     MultiProcessContext,
     MultiProcessTestBase,