run isort

learning-at-home · Jul 13, 2024 · c6ef1f7 · c6ef1f7
1 parent 4ebc04f
commit c6ef1f7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/benchmarks/benchmark_optimizer.py b/benchmarks/benchmark_optimizer.py
@@ -8,6 +8,7 @@
 
 import torch
 import torchvision
+from packaging import version
 from torch import nn as nn
 from torch.nn import functional as F
 from torch.utils.data import Dataset
@@ -16,8 +17,6 @@
 from hivemind.optim.optimizer import Optimizer
 from hivemind.utils.crypto import RSAPrivateKey
 
-from packaging import version
-
 torch_version = torch.__version__.split("+")[0]
 
 if version.parse(torch_version) >= version.parse("2.3.0"):