bohr/heuristics/bugginess/large_change.py

from typing import Optional

from bohrapi.artifacts import Commit
from bohrapi.core import Heuristic
from bohrlabels.core import OneOrManyLabels
from bohrlabels.labels import CommitLabel


@Heuristic(Commit)
def large_change(commit: Commit) -> Optional[OneOrManyLabels]:
    if (
        "bohr" in commit.raw_data
        and "gt_512_codeberta_tokens" in commit.raw_data["bohr"]
    ):
        if commit.raw_data["bohr"]["gt_512_codeberta_tokens"]:
            return CommitLabel.NonBugFix
    return None