Source code for tbp.monty.frameworks.models.salience.strategies.protocol

# Copyright 2026 Thousand Brains Project
#
# Copyright may exist in Contributors' modifications
# and/or contributions to the work.
#
# Use of this source code is governed by the MIT
# license that can be found in the LICENSE file or at
# https://opensource.org/licenses/MIT.
from __future__ import annotations

from typing import Protocol

import numpy as np
import numpy.typing as npt

from tbp.monty.context import RuntimeContext


class SalienceStrategy(Protocol):
    def __call__(
        self,
        ctx: RuntimeContext,
        rgba: npt.NDArray[np.uint8],
        depth: npt.NDArray[np.float64],
    ) -> npt.NDArray[np.float64]: ...