diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index ccddcfcb..cf7fcff3 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -1,8 +1,3 @@ -ci: - autoupdate_schedule: quarterly - -default_stages: [pre-commit] - default_install_hook_types: [pre-commit, commit-msg] repos: @@ -40,3 +35,12 @@ repos: # MD041: first line in a file should be a top-level heading # MD034: bare URL used args: [--disable, MD013, MD033, MD041, MD034, '--'] + + - repo: local + hooks: + - id: ty + name: ty check + entry: ty check + language: python + types: [python] + additional_dependencies: [ty, torch, ase, mace-torch] diff --git a/examples/tutorials/low_level_tutorial.py b/examples/tutorials/low_level_tutorial.py index d0c96a8f..ee31a24f 100644 --- a/examples/tutorials/low_level_tutorial.py +++ b/examples/tutorials/low_level_tutorial.py @@ -67,12 +67,12 @@ model=MaceUrls.mace_mpa_medium, return_raw_model=True, default_dtype=str(dtype).removeprefix("torch."), - device=device, + device=str(device), ) # wrap the mace_mp model in the MaceModel class model = MaceModel( - model=loaded_model, + model=loaded_model, # type: ignore[arg-type] device=device, compute_forces=True, compute_stress=True, diff --git a/pyproject.toml b/pyproject.toml index 1428e67b..6c720a5c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -39,7 +39,6 @@ dependencies = [ test = [ "ase>=3.26", "phonopy>=2.37.0", - "platformdirs>=4.0.0", "psutil>=7.0.0", "pymatgen>=2025.6.14", "pytest-cov>=6", @@ -60,11 +59,11 @@ fairchem = ["fairchem-core>=2.7", "scipy<1.17.0"] docs = [ "autodoc_pydantic==2.2.0", "furo==2024.8.6", - "ipython==8.34.0", "ipykernel==6.30.1", + "ipython==8.34.0", "jsonschema[format]", - "jupyterlab==4.3.4", "jupyter-core==5.8.1", + "jupyterlab==4.3.4", "jupytext==1.16.7", "myst_parser==4.0.0", "nbsphinx>=0.9.7", @@ -148,27 +147,27 @@ conflicts = [ ], [ { extra = "fairchem" }, - { extra = "mace" }, + { extra = "graphpes" }, ], [ - { extra = "graphpes" }, - { extra = "mattersim" }, + { extra = "fairchem" }, + { extra = "mace" }, ], [ - { extra = "graphpes" }, - { extra = "sevenn" }, + { extra = "fairchem" }, + { extra = "mace" }, ], [ { extra = "graphpes" }, - { extra = "fairchem" }, + { extra = "mattersim" }, ], [ { extra = "graphpes" }, { extra = "nequip" }, ], [ - { extra = "fairchem" }, - { extra = "mace" }, + { extra = "graphpes" }, + { extra = "sevenn" }, ], [ { extra = "mace" }, @@ -185,15 +184,13 @@ conflicts = [ ] [dependency-groups] -dev = ["prek>=0.2.0", "ty>=0.0.1a20"] +dev = ["prek>=0.2.13", "ty>=0.0.11"] [tool.ty.rules] -# TODO: Unable to work with **kwargs: https://github.com/astral-sh/ty/issues/247 -missing-argument = "ignore" +unused-ignore-comment = "warn" [[tool.ty.overrides]] include = ["tests/models/**/*.py", "torch_sim/models/**/*.py"] -# TODO would be nice to only ignore unresolved model imports but fail on all other packages [tool.ty.overrides.rules] unresolved-import = "ignore"