fixing typing in dadam

facebookresearch · Oct 9, 2023 · be7a4ff · be7a4ff
1 parent 75426f4
commit be7a4ff
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/audiocraft/optim/dadam.py b/audiocraft/optim/dadam.py
@@ -5,19 +5,15 @@
 # LICENSE file in the root directory of this source tree.
 
 import logging
-from typing import TYPE_CHECKING, Any
+from typing import Any
 
 import torch
 import torch.optim
 import torch.distributed as dist
 
-if TYPE_CHECKING:
-    from torch.optim.optimizer import _params_t
-else:
-    _params_t = Any
-
 
 logger = logging.getLogger(__name__)
+_params_t = Any
 
 
 def to_real(x):