From 7cbbf82cff692a133906746c0738d23e9d998b9c Mon Sep 17 00:00:00 2001
From: hufy <127644049+HuFY-dev@users.noreply.github.com>
Date: Fri, 29 Mar 2024 00:19:43 -0400
Subject: [PATCH] Fixed style issue.

---
 sparse_autoencoder/optimizer/adam_with_reset.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/sparse_autoencoder/optimizer/adam_with_reset.py b/sparse_autoencoder/optimizer/adam_with_reset.py
index dd1921c7..2dee9b15 100644
--- a/sparse_autoencoder/optimizer/adam_with_reset.py
+++ b/sparse_autoencoder/optimizer/adam_with_reset.py
@@ -4,15 +4,16 @@
 """
 from collections.abc import Iterable, Iterator
 from typing import Any
-from typing_extensions import TypeAlias
 
 from jaxtyping import Float, Int
 from torch import Tensor
 from torch.nn.parameter import Parameter
 from torch.optim import Adam
+from typing_extensions import TypeAlias
 
 from sparse_autoencoder.tensor_types import Axis
 
+
 # params_t was renamed to ParamsT in PyTorch 2.2, which caused import errors
 # Copied from PyTorch 2.2 with modifications for better style
 ParamsT: TypeAlias = Iterable[Tensor] | Iterable[dict[str, Any]]