Fix pre-commit

.github/workflows/build_and_publish.yml

@@ -13,7 +13,7 @@ jobs:
     runs-on: ${{ matrix.os }}
     strategy:
       matrix:
-        os: [ ubuntu-20.04 ]
+        os: [ubuntu-20.04]
 
     steps:
       - uses: actions/checkout@v4

.pre-commit-config.yaml

@@ -1,22 +1,52 @@
+exclude: ^(\.copier-answers\.yml)|.pixi$
 repos:
-  - repo: https://github.com/Quantco/pre-commit-mirrors-ruff
-    rev: 0.6.9
+  - repo: local
     hooks:
-      - id: ruff-conda
-      - id: ruff-format-conda
-  - repo: https://github.com/Quantco/pre-commit-mirrors-mypy
-    rev: "1.10.1"
-    hooks:
-      - id: mypy-conda
-        additional_dependencies: [-c, conda-forge, types-setuptools]
-  - repo: https://github.com/Quantco/pre-commit-mirrors-pyupgrade
-    rev: 3.15.2
-    hooks:
-      - id: pyupgrade-conda
-        args:
-          - --py38
-  - repo: https://github.com/Quantco/pre-commit-mirrors-prettier
-    rev: 3.2.5
-    hooks:
-      - id: prettier-conda
-        files: "\\.md$"
+      # ensure pixi environments are up to date
+      # workaround for https://github.com/prefix-dev/pixi/issues/1482
+      - id: pixi-install
+        name: pixi-install
+        entry: pixi install -e default -e lint
+        language: system
+        always_run: true
+        require_serial: true
+        pass_filenames: false
+      # insert-license
+      - id: insert-license
+        name: insert-license
+        entry: >-
+          pixi run -e lint
+          insert-license
+          --license-base64 Q29weXJpZ2h0IChjKSBRdWFudENvIDIwMjMtMjAyNApTUERYLUxpY2Vuc2UtSWRlbnRpZmllcjogQlNELTMtQ2xhdXNlCg==
+          --dynamic-years
+          --comment-style "#"
+        language: system
+        types: [python]
+      # ruff
+      - id: ruff
+        name: ruff
+        entry: pixi run -e lint ruff check --fix --exit-non-zero-on-fix --force-exclude
+        language: system
+        types_or: [python, pyi]
+        require_serial: true
+      - id: ruff-format
+        name: ruff-format
+        entry: pixi run -e lint ruff format --force-exclude
+        language: system
+        types_or: [python, pyi]
+        require_serial: true
+      # mypy
+      - id: mypy
+        name: mypy
+        entry: pixi run -e default mypy
+        language: system
+        types: [python]
+        require_serial: true
+        exclude: ^(tests|api-coverage-tests)/
+      # prettier
+      - id: prettier
+        name: prettier
+        entry: pixi run -e lint prettier --write --list-different --ignore-unknown
+        language: system
+        types: [text]
+        files: \.(md|yml|yaml)$

docs/conf.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 # Configuration file for the Sphinx documentation builder.
 #
 # This file only contains a selection of the most common options. For a full

pixi.toml

@@ -20,6 +20,8 @@ setuptools-scm = "*"
 [feature.lint.dependencies]
 pre-commit = "*"
 insert-license-header = "*"
+mypy = "*"
+prettier = "*"
 
 [feature.lint.tasks]
 pre-commit-install = "pre-commit install"
@@ -45,7 +47,6 @@ pandoc = "*"
 jinja2 = "*"
 python = "3.9.*"
 onnx = "*"
-pre-commit = "*"
 
 [feature.opset-generation.tasks]
 generate-opset = "./tools/generate_opset.sh"
@@ -83,4 +84,4 @@ weekly = ["weekly", "py312", "test"]
 docs = ["docs"]
 build = ["build"]
 lint = ["lint"]
-opset-generation = ["opset-generation"]
+opset-generation = ["opset-generation", "lint"]

src/spox/__init__.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 import importlib.metadata
 
 from spox._public import argument, build, inline

src/spox/_adapt.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 import warnings
 from typing import Dict, List, Optional
 
@@ -41,7 +44,7 @@ def adapt_node(
             for key, var in node.outputs.get_vars().items()
         ]
         initializers = [
-            from_array(var._value, name)
+            from_array(var._value, name)  # type: ignore
             for name, var in node.inputs.get_vars().items()
             if isinstance(var._value, np.ndarray)
         ]

src/spox/_attributes.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 import abc
 from abc import ABC
 from typing import Any, Generic, Iterable, Optional, Tuple, Type, TypeVar, Union

src/spox/_build.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 import itertools
 from dataclasses import dataclass
 from typing import (

src/spox/_debug.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 import sys
 from contextlib import contextmanager
 

src/spox/_exceptions.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 import onnx.shape_inference
 
 InferenceError = onnx.shape_inference.InferenceError

src/spox/_fields.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 import dataclasses
 import enum
 from dataclasses import dataclass

src/spox/_function.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 import inspect
 import itertools
 from dataclasses import dataclass, make_dataclass

src/spox/_future.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 """Module containing experimental Spox features that may be standard in the future."""
 
 from contextlib import contextmanager
@@ -82,7 +85,7 @@ def _promote(
         creating constants and/or casting.
         """
         targets: List[Union[np.dtype, np.generic, int, float]] = [
-            x.type.dtype if isinstance(x, Var) and isinstance(x.type, Tensor) else x
+            x.type.dtype if isinstance(x, Var) and isinstance(x.type, Tensor) else x  # type: ignore
             for x in args
         ]
         if self.type_promotion:

src/spox/_graph.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 """Internal module implementing the low-level Graph object and functions for creating arguments and Graphs."""
 
 import dataclasses

src/spox/_inline.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 import itertools
 from dataclasses import dataclass
 from typing import Callable, Dict, List, Optional, Sequence, Set, Tuple

src/spox/_internal_op.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 """
 Module for Spox operators that implement special internal behaviour that does not fit into the ONNX IR.
 They behave like a normal Node, but their inference, building and translation behaviour may be overriden.

src/spox/_node.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 import dataclasses
 import enum
 import itertools

src/spox/_public.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 """Module implementing the main public interface functions in Spox."""
 
 import contextlib

src/spox/_schemas.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 """Exposes information related to reference ONNX operator schemas, used by StandardOpNode."""
 
 import itertools
@@ -50,11 +53,11 @@ def _current_schema(
 
 def _get_schemas_versioned(
     all_schemas: List[OpSchema],
-) -> Dict[str, Dict[str, OpSchema]]:
+) -> Dict[str, Dict[str, List[OpSchema]]]:
     """Get a map into a list of schemas for all domain/names."""
     return {
         domain: {
-            name: sorted(op_group, key=lambda s: s.since_version)
+            name: sorted(op_group, key=lambda s: s.since_version)  # type: ignore
             for name, op_group in _key_groups(domain_group, lambda s: s.name)
         }
         for domain, domain_group in _key_groups(all_schemas, lambda s: s.domain)
@@ -69,7 +72,7 @@ def _get_schemas_map(
     return {
         domain: {
             version: {
-                name: _current_schema(this_schemas, version)
+                name: _current_schema(this_schemas, version)  # type: ignore
                 for name, this_schemas in domain_schemas.items()
                 if _current_schema(this_schemas, version)
             }
@@ -81,7 +84,7 @@ def _get_schemas_map(
     }
 
 
-ALL_SCHEMAS: List[OpSchema] = get_all_schemas_with_history()
+ALL_SCHEMAS: List[OpSchema] = get_all_schemas_with_history()  # type: ignore
 
 DOMAINS: Set[str] = {s.domain for s in ALL_SCHEMAS}
 

src/spox/_scope.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 from typing import Dict, Generic, Hashable, Optional, Set, TypeVar, Union, overload
 
 from ._node import Node

src/spox/_shape.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 """
 Data representation for Tensor shapes. Largely attempts to mimic semantics of ONNX shapes.
 Shapes have 3 representations:

src/spox/_standard.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 """Module implementing a base for standard ONNX operators, which use the functionality of ONNX node-level inference."""
 
 from typing import TYPE_CHECKING, Callable, Dict, Tuple

src/spox/_traverse.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 from typing import Callable, Iterable, Iterator, List, Optional, Set, Tuple, TypeVar
 
 V = TypeVar("V")

src/spox/_type_system.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 import typing
 from dataclasses import dataclass
 from typing import TypeVar

src/spox/_utils.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 from typing import Optional
 
 import numpy as np

src/spox/_value_prop.py

@@ -1,10 +1,14 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 import enum
 import logging
 import warnings
 from dataclasses import dataclass
-from typing import Dict, List, Union
+from typing import Callable, Dict, List, Union
 
 import numpy as np
+import numpy.typing as npt
 import onnx
 import onnx.reference
 
@@ -147,7 +151,7 @@ def to_ort_value(self) -> ORTValue:
         if self.value is None:  # Optional, Nothing
             return None
         elif isinstance(self.value, PropValue):  # Optional, Some
-            return self.value.to_ref_value()
+            return self.value.to_ref_value()  # type: ignore
         elif isinstance(self.value, list):  # Sequence
             return [elem.to_ref_value() for elem in self.value]
         else:  # Tensor
@@ -192,6 +196,8 @@ def _run_onnxruntime(
 
 
 def get_backend_calls():
+    run: Callable[..., Dict[str, npt.ArrayLike]]
+    unwrap_feed: Callable[..., PropValue]
     if _VALUE_PROP_BACKEND == ValuePropBackend.REFERENCE:
         wrap_feed = PropValue.to_ref_value
         run = _run_reference_implementation

src/spox/_var.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 import typing
 from typing import Any, Callable, ClassVar, Optional, TypeVar, Union
 

src/spox/opset/ai/onnx/ml/v3.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 # ruff: noqa: E741 -- Allow ambiguous variable name
 from dataclasses import dataclass
 from typing import (

src/spox/opset/ai/onnx/ml/v4.py

@@ -1,3 +1,6 @@
+# Copyright (c) QuantCo 2023-2024
+# SPDX-License-Identifier: BSD-3-Clause
+
 # ruff: noqa: E741 -- Allow ambiguous variable name
 from dataclasses import dataclass
 from typing import (