From d2bcdbff5e01d8191748bd67f9f7a89b33446e22 Mon Sep 17 00:00:00 2001
From: Duo <50307526+iProzd@users.noreply.github.com>
Date: Thu, 9 May 2024 14:30:02 +0800
Subject: [PATCH] Fix typo

---
 deepmd/dpmodel/descriptor/descriptor.py  | 2 +-
 deepmd/dpmodel/descriptor/dpa1.py        | 2 +-
 deepmd/dpmodel/descriptor/repformers.py  | 2 +-
 deepmd/pt/model/descriptor/descriptor.py | 2 +-
 deepmd/pt/model/descriptor/se_atten.py   | 2 +-
 5 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/deepmd/dpmodel/descriptor/descriptor.py b/deepmd/dpmodel/descriptor/descriptor.py
index f11f406399..444df1abf8 100644
--- a/deepmd/dpmodel/descriptor/descriptor.py
+++ b/deepmd/dpmodel/descriptor/descriptor.py
@@ -71,7 +71,7 @@ def get_dim_out(self) -> int:
 
     @abstractmethod
     def get_dim_in(self) -> int:
-        """Returns the output dimension."""
+        """Returns the input dimension."""
         pass
 
     @abstractmethod
diff --git a/deepmd/dpmodel/descriptor/dpa1.py b/deepmd/dpmodel/descriptor/dpa1.py
index 9d1dee2481..e5e79a8984 100644
--- a/deepmd/dpmodel/descriptor/dpa1.py
+++ b/deepmd/dpmodel/descriptor/dpa1.py
@@ -648,7 +648,7 @@ def get_ntypes(self) -> int:
         return self.ntypes
 
     def get_dim_in(self) -> int:
-        """Returns the output dimension."""
+        """Returns the input dimension."""
         return self.dim_in
 
     def get_dim_out(self) -> int:
diff --git a/deepmd/dpmodel/descriptor/repformers.py b/deepmd/dpmodel/descriptor/repformers.py
index bd2f97a29a..ca3651a979 100644
--- a/deepmd/dpmodel/descriptor/repformers.py
+++ b/deepmd/dpmodel/descriptor/repformers.py
@@ -268,7 +268,7 @@ def get_ntypes(self) -> int:
         return self.ntypes
 
     def get_dim_in(self) -> int:
-        """Returns the output dimension."""
+        """Returns the input dimension."""
         return self.dim_in
 
     def get_dim_out(self) -> int:
diff --git a/deepmd/pt/model/descriptor/descriptor.py b/deepmd/pt/model/descriptor/descriptor.py
index 5aae848aa4..d586fc988f 100644
--- a/deepmd/pt/model/descriptor/descriptor.py
+++ b/deepmd/pt/model/descriptor/descriptor.py
@@ -80,7 +80,7 @@ def get_dim_out(self) -> int:
 
     @abstractmethod
     def get_dim_in(self) -> int:
-        """Returns the output dimension."""
+        """Returns the input dimension."""
         pass
 
     @abstractmethod
diff --git a/deepmd/pt/model/descriptor/se_atten.py b/deepmd/pt/model/descriptor/se_atten.py
index 465bbb8308..958f3b4963 100644
--- a/deepmd/pt/model/descriptor/se_atten.py
+++ b/deepmd/pt/model/descriptor/se_atten.py
@@ -299,7 +299,7 @@ def get_ntypes(self) -> int:
         return self.ntypes
 
     def get_dim_in(self) -> int:
-        """Returns the output dimension."""
+        """Returns the input dimension."""
         return self.dim_in
 
     def get_dim_out(self) -> int: