unused import [5/n] (#3413)

Summary:
Pull Request resolved: #3413

https://www.flake8rules.com/rules/F401.html

Reviewed By: TroyGarden

Differential Revision: D83739291

fbshipit-source-id: 37fb0570f0e82b34aa7324f2054cad316f0674ee

Author: aliafzal

torchrec/distributed/benchmark/benchmark_zch/benchmark_zch.py

@@ -9,13 +9,12 @@
 import argparse
 import csv
 import json
-import logging
 import multiprocessing
 import os
 import sys
 import time
 
-from typing import cast, Dict, Iterator, List, Optional
+from typing import Dict, List, Optional
 
 import numpy as np
 

torchrec/distributed/planner/planners.py

@@ -14,8 +14,6 @@
 from time import perf_counter
 from typing import Callable, cast, Dict, List, Optional, Tuple, Union
 
-import torch
-
 import torch.distributed as dist
 from torch import nn
 from torchrec.distributed.collective_utils import invoke_on_rank_and_broadcast_result
@@ -32,7 +30,7 @@
     GridSearchProposer,
     UniformProposer,
 )
-from torchrec.distributed.planner.stats import EmbeddingStats, NoopEmbeddingStats
+from torchrec.distributed.planner.stats import EmbeddingStats
 from torchrec.distributed.planner.storage_reservations import (
     HeuristicalStorageReservation,
 )

torchrec/distributed/sharding/twrw_sharding.py

@@ -9,7 +9,7 @@
 
 import itertools
 import math
-from typing import Any, cast, Dict, List, Optional, Tuple, TypeVar, Union
+from typing import Any, cast, Dict, List, Optional, Tuple, TypeVar
 
 import torch
 import torch.distributed as dist