Source code for xmm.pipeline.steps.filter

from xmm.util.functions import get_function_from_str
from .base import PipelineStep


class FilterStep(PipelineStep):
    """Filter rows."""

    def __init__(self, filter_func):
        self.filter_func = get_function_from_str(filter_func)

    def process_step(self, state, context):
        filtered = filter(self.filter_func, state)

        return filtered, context