Class OperatorStats

datastudio/utils/statistics.py:18–54

Statistics for a single operator execution.

@dataclass
class OperatorStats:
    """Statistics for a single operator execution."""

    name: str
    input_count: int = 0
    output_count: int = 0
    rejected_count: int = 0
    rewritten_count: int = 0
    execution_time: float = 0.0
    # QA-level statistics
    input_qa_count: int = 0
    output_qa_count: int = 0
    rejected_qa_count: int = 0
    # Split statistics (partial rejection causes item split)
    split_count: int = 0
    details: Dict[str, Any] = field(default_factory=dict)

    @property
    def reject_rate(self) -> float:
        """Calculate reject rate (item level)."""
        if self.input_count == 0:
            return 0.0
        return self.rejected_count / self.input_count

    @property
    def rewrite_rate(self) -> float:
        """Calculate rewrite rate."""
        if self.input_count == 0:
            return 0.0
        return self.rewritten_count / self.input_count

    @property
    def qa_reject_rate(self) -> float:
        """Calculate QA-level reject rate."""
        if self.input_qa_count == 0:
            return 0.0
        return self.rejected_qa_count / self.input_qa_count


@dataclass