fix: import

kozistr · Nov 23, 2024 · 90d43d4 · 90d43d4
1 parent 7a603a1
commit 90d43d4
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/tests/test_optimizer_parameters.py b/tests/test_optimizer_parameters.py
@@ -2,16 +2,8 @@
 import torch
 from torch import nn
 
-from pytorch_optimizer.optimizer import (
-    SAM,
-    WSAM,
-    GaLoreProjector,
-    Lookahead,
-    PCGrad,
-    Ranger21,
-    SafeFP16Optimizer,
-    load_optimizer,
-)
+from pytorch_optimizer.optimizer import SAM, WSAM, Lookahead, PCGrad, Ranger21, SafeFP16Optimizer, load_optimizer
+from pytorch_optimizer.optimizer.galore import GaLoreProjector
 from tests.constants import PULLBACK_MOMENTUM
 from tests.utils import Example, simple_parameter, simple_zero_rank_parameter