fix: adjust client ruff settings

This commit is contained in:
drbh 2024-07-19 17:33:11 +00:00
parent 154cf67dad
commit 655a9d7ef3

View File

@ -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."
)