task-sdk/.pre-commit-config.yaml

# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.
---
default_stages: [pre-commit, pre-push]
minimum_prek_version: '0.3.4'
default_language_version:
  python: python3
  node: 22.19.0
  golang: 1.24.0
repos:
  - repo: local
    hooks:
      - id: check-core-imports
        name: Check for core imports in task-sdk files
        entry: ../scripts/ci/prek/check_core_imports_in_sdk.py
        language: python
        types: [python]
        files: ^src/airflow/.*\.py$
        exclude: |
          (?x)
          # TODO: These files need to be refactored to remove core coupling
          ^src/airflow/sdk/bases/operator\.py$|
          ^src/airflow/sdk/definitions/decorators/__init__\.pyi$|
          ^src/airflow/sdk/definitions/decorators/setup_teardown\.py$|
          ^src/airflow/sdk/definitions/asset/__init__\.py$|
          ^src/airflow/sdk/definitions/asset/decorators\.py$|
          ^src/airflow/sdk/definitions/taskgroup\.py$|
          ^src/airflow/sdk/definitions/mappedoperator\.py$|
          ^src/airflow/sdk/definitions/deadline\.py$|
          ^src/airflow/sdk/definitions/dag\.py$|
          ^src/airflow/sdk/definitions/_internal/types\.py$|
          ^src/airflow/sdk/execution_time/callback_supervisor\.py$|
          ^src/airflow/sdk/execution_time/execute_workload\.py$|
          ^src/airflow/sdk/execution_time/secrets_masker\.py$|
          ^src/airflow/sdk/execution_time/connection_test_supervisor\.py$|
          ^src/airflow/sdk/execution_time/schema/__init__\.py$|
          ^src/airflow/sdk/execution_time/supervisor\.py$|
          ^src/airflow/sdk/execution_time/task_runner\.py$|
          ^src/airflow/sdk/serde/serializers/kubernetes\.py$|
          ^src/airflow/sdk/types.py$
      - id: check-init-decorator-arguments
        name: Sync model __init__ and decorator arguments
        language: python
        entry: ../scripts/ci/prek/check_init_decorator_arguments.py
        pass_filenames: false
        files: ^src/airflow/sdk/definitions/dag\.py$|^src/airflow/sdk/definitions/decorators/task_group\.py$
      - id: mypy-task-sdk
        name: Run mypy for task-sdk
        language: python
        entry: ../scripts/ci/prek/run_mypy_full_dist_local_venv_or_breeze_in_ci.py task-sdk
        pass_filenames: false
        files: ^.*\.py$
        require_serial: true