From e89840768b682fb3c48328b523fd7ee23b41a4b1 Mon Sep 17 00:00:00 2001 From: Michael Benayoun Date: Fri, 16 Sep 2022 11:41:53 +0200 Subject: [PATCH] Update BaseConfig.get_configuration_file docstring to match transformers --- optimum/configuration_utils.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/optimum/configuration_utils.py b/optimum/configuration_utils.py index 3e752e2554..8ffeda45a5 100644 --- a/optimum/configuration_utils.py +++ b/optimum/configuration_utils.py @@ -1,3 +1,4 @@ +# coding=utf-8 # Copyright 2021 The HuggingFace Team. All rights reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); @@ -95,8 +96,10 @@ def save_pretrained(self, save_directory: Union[str, os.PathLike], push_to_hub: def get_configuration_file(cls, configuration_files: List[str]) -> str: """ Get the configuration file to use for this version of transformers. + Args: configuration_files (`List[str]`): The list of available configuration files. + Returns: `str`: The configuration file to use. """ @@ -109,6 +112,7 @@ def get_configuration_file(cls, configuration_files: List[str]) -> str: configuration_files_map[v] = file_name available_versions = sorted(configuration_files_map.keys()) + # Defaults to FULL_CONFIGURATION_FILE and then try to look at some newer versions. configuration_file = cls.CONFIG_NAME optimum_version = version.parse(__version__) for v in available_versions: