precommit

mosaicml · Dec 7, 2023 · 0e29f8a · 0e29f8a
1 parent f48ab24
commit 0e29f8a
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/tests/optim/test_lion8b.py b/tests/optim/test_lion8b.py
@@ -6,14 +6,12 @@
 import warnings
 
 import numpy as np
-import packaging.version as version
 import pytest
 import torch
 import torch.distributed as dist
 import torch.nn as nn
 from torch.distributed import fsdp
 from torch.distributed.fsdp import FullyShardedDataParallel as FSDP
-
 from torch.distributed.fsdp.api import (  # type:ignore .api not in public API
     FullOptimStateDictConfig, LocalOptimStateDictConfig,
     ShardedOptimStateDictConfig)