From 9681276fb81684a6f97db73cdf10446ac9844958 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Mon, 26 Feb 2024 03:18:24 -0500 Subject: [PATCH] Apply suggestions from code review Signed-off-by: Jinzhe Zeng --- deepmd/pt/model/model/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/deepmd/pt/model/model/model.py b/deepmd/pt/model/model/model.py index 000c6c6de6..166b1299fe 100644 --- 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