This commit is contained in:
William FH 2023-08-03 09:33:28 -07:00 committed by GitHub
parent 1f54ec899b
commit 8022293124
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -127,11 +127,12 @@ class ChainStringRunMapper(StringRunMapper):
input_key: Optional[str] = None input_key: Optional[str] = None
"""The key from the model Run's inputs to use as the eval input. """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 prediction_key: Optional[str] = None
"""The key from the model Run's outputs to use as the eval prediction. """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: def _get_key(self, source: Dict, key: Optional[str], which: str) -> str:
if key is not None: if key is not None: