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 12, 2024
1 parent 159c4c6 commit 1f8c74c
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 22 deletions.
6 changes: 3 additions & 3 deletions deepmd/pt/model/model/dp_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,16 +18,16 @@
from deepmd.pt.model.task.dipole import (
DipoleFittingNet,
)
from deepmd.pt.model.task.dos import (
DOSFittingNet,
)
from deepmd.pt.model.task.ener import (
EnergyFittingNet,
EnergyFittingNetDirect,
)
from deepmd.pt.model.task.polarizability import (
PolarFittingNet,
)
from deepmd.pt.model.task.dos import (
DOSFittingNet
)

from .make_model import (
make_model,
Expand Down
19 changes: 0 additions & 19 deletions deepmd/pt/model/task/dos.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,24 +2,12 @@
import copy
import logging
from typing import (

Check warning on line 4 in deepmd/pt/model/task/dos.py

View check run for this annotation

Codecov / codecov/patch

deepmd/pt/model/task/dos.py#L2-L4

Added lines #L2 - L4 were not covered by tests
Callable,
List,
Optional,
Tuple,
Union,
)

import numpy as np
import torch

Check warning on line 9 in deepmd/pt/model/task/dos.py

View check run for this annotation

Codecov / codecov/patch

deepmd/pt/model/task/dos.py#L9

Added line #L9 was not covered by tests

from deepmd.dpmodel import (
FittingOutputDef,
OutputVariableDef,
fitting_check_output,
)
from deepmd.pt.model.network.network import (
ResidualDeep,
)
from deepmd.pt.model.task.fitting import (

Check warning on line 11 in deepmd/pt/model/task/dos.py

View check run for this annotation

Codecov / codecov/patch

deepmd/pt/model/task/dos.py#L11

Added line #L11 was not covered by tests
InvarFitting,
)
Expand All @@ -29,12 +17,6 @@
from deepmd.pt.utils.env import (

Check warning on line 17 in deepmd/pt/model/task/dos.py

View check run for this annotation

Codecov / codecov/patch

deepmd/pt/model/task/dos.py#L17

Added line #L17 was not covered by tests
DEFAULT_PRECISION,
)
from deepmd.pt.utils.stat import (
compute_output_stats,
)
from deepmd.utils.path import (
DPPath,
)
from deepmd.utils.version import (

Check warning on line 20 in deepmd/pt/model/task/dos.py

View check run for this annotation

Codecov / codecov/patch

deepmd/pt/model/task/dos.py#L20

Added line #L20 was not covered by tests
check_version_compatibility,
)
Expand Down Expand Up @@ -102,4 +84,3 @@ def serialize(self) -> dict:

# make jit happy with torch 2.0.0
exclude_types: List[int]

Check warning on line 86 in deepmd/pt/model/task/dos.py

View check run for this annotation

Codecov / codecov/patch

deepmd/pt/model/task/dos.py#L86

Added line #L86 was not covered by tests

0 comments on commit 1f8c74c

Please sign in to comment.