Spaces:
Runtime error
Runtime error
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() | |