From 3bddb4a6f76fe5a40553ee9e692f6120cf29bd13 Mon Sep 17 00:00:00 2001
From: Lee Harrold <35541778+Harrolee@users.noreply.github.com>
Date: Sat, 20 Jul 2024 11:42:19 -0400
Subject: [PATCH] Update param_t to ParamsT

The type changed
---
 sparse_autoencoder/optimizer/adam_with_reset.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/sparse_autoencoder/optimizer/adam_with_reset.py b/sparse_autoencoder/optimizer/adam_with_reset.py
index 5a84c941..ae906efd 100644
--- a/sparse_autoencoder/optimizer/adam_with_reset.py
+++ b/sparse_autoencoder/optimizer/adam_with_reset.py
@@ -8,7 +8,7 @@
 from torch import Tensor
 from torch.nn.parameter import Parameter
 from torch.optim import Adam
-from torch.optim.optimizer import params_t
+from torch.optim.optimizer import ParamsT
 
 from sparse_autoencoder.tensor_types import Axis
 
@@ -35,7 +35,7 @@ class AdamWithReset(Adam):
 
     def __init__(  # (extending existing implementation)
         self,
-        params: params_t,
+        params: ParamsT,
         lr: float | Float[Tensor, Axis.names(Axis.SINGLE_ITEM)] = 1e-3,
         betas: tuple[float, float] = (0.9, 0.999),
         eps: float = 1e-8,