remove deprecated extension

masci · Nov 10, 2024 · 2635408 · 2635408
1 parent 32c3841
commit 2635408
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 62 deletions.
diff --git a/src/banks/env.py b/src/banks/env.py
@@ -16,12 +16,10 @@ def _add_extensions(_env):
     from .extensions.chat import ChatExtension  # pylint: disable=import-outside-toplevel
     from .extensions.completion import CompletionExtension  # pylint: disable=import-outside-toplevel
     from .extensions.generate import GenerateExtension  # pylint: disable=import-outside-toplevel
-    from .extensions.inference_endpoint import HFInferenceEndpointsExtension  # pylint: disable=import-outside-toplevel
 
     _env.add_extension(ChatExtension)
     _env.add_extension(CompletionExtension)
     _env.add_extension(GenerateExtension)
-    _env.add_extension(HFInferenceEndpointsExtension)
 
 
 # Init the Jinja env

diff --git a/src/banks/extensions/__init__.py b/src/banks/extensions/__init__.py
@@ -2,6 +2,5 @@
 #
 # SPDX-License-Identifier: MIT
 from banks.extensions.generate import GenerateExtension
-from banks.extensions.inference_endpoint import HFInferenceEndpointsExtension
 
-__all__ = ("GenerateExtension", "HFInferenceEndpointsExtension")
+__all__ = ("GenerateExtension",)
diff --git a/src/banks/extensions/inference_endpoint.py b/src/banks/extensions/inference_endpoint.py