OpenFactCheck-Prerelease / src /openfactcheck /solvers /dummy /useless_response_regenerator.py
Hasan Iqbal
Made the imports more user friendly
ec53a03 unverified
raw
history blame contribute delete
569 Bytes
from openfactcheck import FactCheckerState, StandardTaskSolver, Solver
@Solver.register("useless_response_regenerator", "claims_with_tags", "output")
class UselessResponseRegenerator(StandardTaskSolver):
def __init__(self, args):
super().__init__(args)
def __call__(self, state: FactCheckerState, *args, **kwargs):
claims = state.get(self.input_name)
true_claims = [k[1] for k, v in claims.items() if v is True]
new_response = ' '.join(true_claims)
state.set(self.output_name, new_response)
return True, state