| from compliance_checks.base import ComplianceResult, ComplianceCheck, walk_to_next_heading | |
| from bs4 import BeautifulSoup | |
| class GeneralLimitationsResult(ComplianceResult): | |
| name = "General Limitations" | |
| def __init__( | |
| self, | |
| limitations: str = None, | |
| *args, | |
| **kwargs, | |
| ): | |
| super().__init__(*args, **kwargs) | |
| self.limitations = limitations | |
| def __eq__(self, other): | |
| if isinstance(other, GeneralLimitationsResult): | |
| if super().__eq__(other): | |
| try: | |
| assert self.limitations == other.limitations | |
| return True | |
| except AssertionError: | |
| return False | |
| else: | |
| return False | |
| def to_string(self): | |
| return self.limitations | |
| class GeneralLimitationsCheck(ComplianceCheck): | |
| name = "General Limitations" | |
| def run_check(self, card: BeautifulSoup): | |
| combos = [ | |
| ("h1", "Bias, Risks, and Limitations"), ("h2", "Bias, Risks, and Limitations"), | |
| ("h2", "Intended uses & limitations"), | |
| ("h1", "Risks and Limitations"), | |
| ("h2", "Risks, Limitations and Biases"), | |
| ("h2", "Limitations and Bias"), | |
| ("h3", "Limitations and bias"), | |
| ("h2", "Limitations"), | |
| ] | |
| for hX, heading in combos: | |
| purpose_check = walk_to_next_heading(card, hX, heading) | |
| if purpose_check: | |
| return GeneralLimitationsResult( | |
| status=True, | |
| ) | |
| return GeneralLimitationsResult() | |