From f67fa2189c59dc7819a7bb2a0d06cae9696d8bbd Mon Sep 17 00:00:00 2001
From: Han Wang <wang_han@iapcm.ac.cn>
Date: Tue, 9 Jan 2024 23:04:41 +0800
Subject: [PATCH] fix unused var

---
 deepmd_utils/model_format/se_e2_a.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deepmd_utils/model_format/se_e2_a.py b/deepmd_utils/model_format/se_e2_a.py
index 390abffa30..114f9df915 100644
--- a/deepmd_utils/model_format/se_e2_a.py
+++ b/deepmd_utils/model_format/se_e2_a.py
@@ -73,7 +73,7 @@ def __init__(
         for ii in range(self.ntypes):
             self.embeddings.append(
                 EmbeddingNet(
-                    1,
+                    in_dim,
                     self.neuron,
                     self.activation_function,
                     self.resnet_dt,