Unverified Commit c9d2cf85 authored by Alexander Measure's avatar Alexander Measure Committed by GitHub
Browse files

add Tuple as possible type hint for EvalPredictions label_ids (#14473)

* Update trainer_utils.py

* add Tuple type hints to all label_ids outputs

affects EvalLoopOutput and PredicctionOutput
parent a59e7c1e
...@@ -75,19 +75,19 @@ class EvalPrediction(NamedTuple): ...@@ -75,19 +75,19 @@ class EvalPrediction(NamedTuple):
""" """
predictions: Union[np.ndarray, Tuple[np.ndarray]] predictions: Union[np.ndarray, Tuple[np.ndarray]]
label_ids: np.ndarray label_ids: Union[np.ndarray, Tuple[np.ndarray]]
class EvalLoopOutput(NamedTuple): class EvalLoopOutput(NamedTuple):
predictions: Union[np.ndarray, Tuple[np.ndarray]] predictions: Union[np.ndarray, Tuple[np.ndarray]]
label_ids: Optional[np.ndarray] label_ids: Optional[Union[np.ndarray, Tuple[np.ndarray]]]
metrics: Optional[Dict[str, float]] metrics: Optional[Dict[str, float]]
num_samples: Optional[int] num_samples: Optional[int]
class PredictionOutput(NamedTuple): class PredictionOutput(NamedTuple):
predictions: Union[np.ndarray, Tuple[np.ndarray]] predictions: Union[np.ndarray, Tuple[np.ndarray]]
label_ids: Optional[np.ndarray] label_ids: Optional[Union[np.ndarray, Tuple[np.ndarray]]]
metrics: Optional[Dict[str, float]] metrics: Optional[Dict[str, float]]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment