fix: typo

deepmodeling · Mar 11, 2024 · 9ad2c97 · 9ad2c97
1 parent 042c86b
commit 9ad2c97
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/pt/model/task/polarizability.py b/deepmd/pt/model/task/polarizability.py
@@ -268,7 +268,7 @@ def compute_output_stats(
                 # handle nan values.
                 constant_matrix = np.nan_to_num(constant_matrix)
             if stat_file_path is not None:
-                stat_file_path.save_numpy(self.constant_matrix)
+                stat_file_path.save_numpy(constant_matrix)
             self.constant_matrix = torch.tensor(constant_matrix, device=env.DEVICE)
 
     def forward(