Apply suggestions from code review

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Feb 26, 2024 · 9681276 · 9681276
1 parent 4a17c1f
commit 9681276
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepmd/pt/model/model/model.py b/deepmd/pt/model/model/model.py
@@ -26,7 +26,7 @@
 # the abstract method will override the method from the atomic model
 # as Python resolves method lookups using the C3 linearisation.
 # See https://stackoverflow.com/a/47117600/9567349
-# Take an example, this is sitatuion for only inherbiting BaseBaseModel:
+# Take an example, this is the situation for only inheriting BaseBaseModel:
 #       torch.nn.Module        BaseAtomicModel        BaseBaseModel
 #             |                       |                    |
 #             -------------------------                    |
@@ -40,7 +40,7 @@
 #                                         DPModel
 #
 # The order is: DPModel -> make_model(DPAtomicModel) -> DPAtomicModel ->
-# torch.nn.Module -> BaseAtomicModel -> BaseBaseModel
+# torch.nn.Module -> BaseAtomicModel -> BaseModel -> BaseBaseModel
 #
 # However, if BaseModel also inherbits from torch.nn.Module:
 #         torch.nn.Module                      BaseBaseModel