Limit imports

meltano · Aug 5, 2024 · a45b0cb · a45b0cb
1 parent 2543c77
commit a45b0cb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/...ecutter/tap-template/{{cookiecutter.tap_id}}/{{cookiecutter.library_name}}/rest-client.py b/...ecutter/tap-template/{{cookiecutter.tap_id}}/{{cookiecutter.library_name}}/rest-client.py
@@ -47,7 +47,11 @@
 
 if TYPE_CHECKING:
     import requests
+    {%- if cookiecutter.auth_method in ("OAuth2", "JWT") %}
     from singer_sdk.helpers.types import Auth, Context
+    {%- else %}
+    from singer_sdk.helpers.types import Context
+    {%- endif %}
 
 
 # TODO: Delete this is if not using json files for schema definition