From 9c7a61afca0cdd3392888feb62bb534f0fb9d3d2 Mon Sep 17 00:00:00 2001
From: 152334H <54623771+152334H@users.noreply.github.com>
Date: Fri, 9 Feb 2024 17:39:38 +0800
Subject: [PATCH] fix import renaming on python 2.2.0

---
 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 5a84c941..dba88c66 100644
--- a/sparse_autoencoder/optimizer/adam_with_reset.py
+++ b/sparse_autoencoder/optimizer/adam_with_reset.py
@@ -8,7 +8,8 @@
 from torch import Tensor
 from torch.nn.parameter import Parameter
 from torch.optim import Adam
-from torch.optim.optimizer import params_t
+try: from torch.optim.optimizer import params_t
+except ImportError: from torch.optim.optimizer import ParamsT as params_t
 
 from sparse_autoencoder.tensor_types import Axis