From fa93b96bcd8358821ca7dfe04dc881b0833e86e6 Mon Sep 17 00:00:00 2001 From: sami jaghouar Date: Sat, 23 Aug 2025 04:23:41 +0000 Subject: [PATCH] remove unused import --- torchft/local_sgd.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/torchft/local_sgd.py b/torchft/local_sgd.py index 5129e670..e7270992 100644 --- a/torchft/local_sgd.py +++ b/torchft/local_sgd.py @@ -10,18 +10,14 @@ """ import logging import math -import threading from contextlib import nullcontext from types import TracebackType -from typing import Any, Callable, Dict, Iterator, List, Optional, Tuple, Type +from typing import Any, Dict, List, Optional, Tuple, Type import torch -import torch.distributed as dist from torch import nn, optim from torch.distributed.distributed_c10d import Work from torch.distributed.tensor import DTensor -from torch.nn.parameter import Parameter -from torch.optim.optimizer import Optimizer from torch.utils.hooks import RemovableHandle from torchft.manager import Manager