From 655a9d7ef3a32fb3ce1e77966504e5daf0331ba8 Mon Sep 17 00:00:00 2001 From: drbh Date: Fri, 19 Jul 2024 17:33:11 +0000 Subject: [PATCH] fix: adjust client ruff settings --- clients/python/text_generation/__init__.py | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/clients/python/text_generation/__init__.py b/clients/python/text_generation/__init__.py index 8444a08e..ca783dcd 100644 --- a/clients/python/text_generation/__init__.py +++ b/clients/python/text_generation/__init__.py @@ -12,12 +12,18 @@ # See the License for the specific language governing permissions and # limitations under the License. -from text_generation.client import Client, AsyncClient -from text_generation.inference_api import InferenceAPIClient, InferenceAPIAsyncClient - - __version__ = "0.7.0" +DEPRECATION_WARNING = ( + "`text_generation` clients are deprecated and will be removed in the near future. " + "Please use the `InferenceClient` from the `huggingface_hub` package instead." +) + +from text_generation.client import Client, AsyncClient # noqa E402 +from text_generation.inference_api import ( # noqa E402 + InferenceAPIClient, + InferenceAPIAsyncClient, +) __all__ = [ "Client", @@ -25,8 +31,3 @@ __all__ = [ "InferenceAPIClient", "InferenceAPIAsyncClient", ] - -DEPRECATION_WARNING = ( - "`text_generation` clients are deprecated and will be removed in the near future. " - "Please use the `InferenceClient` from the `huggingface_hub` package instead." -)