fix: Changed call location

AI21Labs · Jun 24, 2024 · 502f9c0 · 502f9c0
1 parent e647583
commit 502f9c0
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/ai21/__init__.py b/ai21/__init__.py
@@ -18,6 +18,7 @@
 from ai21.version import VERSION
 
 __version__ = VERSION
+setup_logger()
 
 
 def _import_bedrock_client():
@@ -69,5 +70,3 @@ def __getattr__(name: str) -> Any:
     "AI21AzureClient",
     "AsyncAI21AzureClient",
 ]
-
-setup_logger()