Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Mar 1, 2024
1 parent bb757c3 commit 7747cc4
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions doc/development/create-a-model.md
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,7 @@ from deepmd.pt.model.descriptor.base_descriptor import (
BaseDescriptor,
)


@BaseDescriptor.register("some_descrpt")
class SomeDescript(BaseDescriptor, torch.nn.Module):
def __init__(self, arg1: bool, arg2: float) -> None:
Expand Down Expand Up @@ -72,6 +73,7 @@ from deepmd.dpmodel import (
fitting_check_output,
)


@GeneralFitting.register("some_fitting")
@fitting_check_output
class SomeFittingNet(GeneralFitting):
Expand Down Expand Up @@ -111,17 +113,20 @@ from deepmd.pt.model.model.model import (
BaseModel,
)


class SomeAtomicModel(BaseAtomicModel, torch.nn.Module):
def __init__(self, arg1: bool, arg2: float) -> None:
pass

def forward_atomic(self):
pass


@BaseModel.register("some_model")
class SomeDPModel(make_model(SomeAtomicModel), BaseModel):
pass


class SomeModel(SomeDPModel):
pass
```
Expand Down

0 comments on commit 7747cc4

Please sign in to comment.