diff --git a/datasets_old_2022_10/interface.py b/datasets_old_2022_10/interface.py index ce925b7a..b3e126e6 100644 --- a/datasets_old_2022_10/interface.py +++ b/datasets_old_2022_10/interface.py @@ -26,19 +26,19 @@ def __repr__(self): parts.insert(0, f"class={ds['class']}") return f"<{self.__class__.__name__} {' '.join(parts)}>" - def get_extern_data(self) -> Dict[str, Dict[str]]: + def get_extern_data(self) -> Dict[str, Dict[str, Any]]: """ Get extern data """ raise NotImplementedError - def get_train_dataset(self) -> Dict[str]: + def get_train_dataset(self) -> Dict[str, Any]: """ Get train dataset """ raise NotImplementedError - def get_eval_datasets(self) -> Dict[str, Dict[str]]: + def get_eval_datasets(self) -> Dict[str, Dict[str, Any]]: """ :return: e.g. {"dev": ..., "devtrain": ...} This is intended for eval_datasets in the RETURNN config, @@ -74,7 +74,7 @@ def get_default_target(self) -> Optional[str]: """ return "classes" - def get_config_opts(self) -> Dict[str]: + def get_config_opts(self) -> Dict[str, Any]: """ E.g. in your main config, you could do::