This commit is contained in:
William FH
2023-08-03 09:33:28 -07:00
committed by GitHub
parent 1f54ec899b
commit 8022293124

View File

@@ -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: