Skip to content

Commit bb97505

Browse files
committed
Update HuggingFaceEvaluation to support compute args.
1 parent b5bcb8d commit bb97505

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ads/llm/guardrails/huggingface.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
class HuggingFaceEvaluation(Guardrail):
1414
path: str = ""
1515
load_args: dict = {}
16+
compute_args: dict = {}
1617
_evaluator: evaluate.EvaluationModule = ""
1718

1819
@root_validator(skip_on_failure=True)
@@ -36,7 +37,7 @@ def compute(self, data=None, **kwargs):
3637
load_args = {"path": self.path}
3738
load_args.update(self.load_args)
3839
self._evaluator = evaluate.load(**load_args)
39-
return self._evaluator.compute(predictions=data, **kwargs)
40+
return self._evaluator.compute(predictions=data, **self.compute_args, **kwargs)
4041

4142
@property
4243
def metric_key(self):

0 commit comments

Comments
 (0)