Source code for todd.models.filters.base

__all__ = [
    'BaseFilter',
]

from abc import ABC, abstractmethod
from typing import Generator, Generic, TypeVar

from torch import nn

T = TypeVar('T')




class BaseFilter(Generic[T], ABC):

    @abstractmethod
    def __call__(self, module: nn.Module) -> Generator[T, None, None]:
        pass