pull/8705/head
William FH 1 year ago committed by GitHub
parent 1f54ec899b
commit 8022293124
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -127,11 +127,12 @@ class ChainStringRunMapper(StringRunMapper):
input_key: Optional[str] = None
"""The key from the model Run's inputs to use as the eval input.
If not provided, will use the only input key or raise an error if there are multiple.
"""
If not provided, will use the only input key or raise an
error if there are multiple."""
prediction_key: Optional[str] = None
"""The key from the model Run's outputs to use as the eval prediction.
If not provided, will use the only output key or raise an error if there are multiple."""
If not provided, will use the only output key or raise an error
if there are multiple."""
def _get_key(self, source: Dict, key: Optional[str], which: str) -> str:
if key is not None:

Loading…
Cancel
Save