Skip to content

Commit

Permalink
Fix ModelCardData's datasets typing
Browse files Browse the repository at this point in the history
  • Loading branch information
hanouticelina committed Oct 29, 2024
1 parent 786d3ba commit b0ce065
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/huggingface_hub/repocard_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -252,8 +252,8 @@ class ModelCardData(CardData):
The identifier of the base model from which the model derives. This is applicable for example if your model is a
fine-tune or adapter of an existing model. The value must be the ID of a model on the Hub (or a list of IDs
if your model derives from multiple models). Defaults to None.
datasets (`List[str]`, *optional*):
List of datasets that were used to train this model. Should be a dataset ID
datasets (`Union[str, List[str]]`, *optional*):
Dataset or list of datasets that were used to train this model. Should be a dataset ID
found on https://hf.co/datasets. Defaults to None.
eval_results (`Union[List[EvalResult], EvalResult]`, *optional*):
List of `huggingface_hub.EvalResult` that define evaluation results of the model. If provided,
Expand Down Expand Up @@ -312,7 +312,7 @@ def __init__(
self,
*,
base_model: Optional[Union[str, List[str]]] = None,
datasets: Optional[List[str]] = None,
datasets: Optional[Union[str, List[str]]] = None,
eval_results: Optional[List[EvalResult]] = None,
language: Optional[Union[str, List[str]]] = None,
library_name: Optional[str] = None,
Expand Down

0 comments on commit b0ce065

Please sign in to comment.