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