fix lint for python 3.8

statsig-io · Nov 7, 2024 · 73a1ce8 · 73a1ce8
1 parent c9ee5a4
commit 73a1ce8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/azureai/get_statsig_user.py b/azureai/get_statsig_user.py
@@ -6,5 +6,5 @@ def get_statsig_user(user: Optional[StatsigUser] = None) -> StatsigUser:
     if user is not None:
         user.custom_ids["sdk_type"] = "azureai-python"
         return user
-    else:
-        return StatsigUser(custom_ids={"sdk_type": "azureai-python"})
+
+    return StatsigUser(custom_ids={"sdk_type": "azureai-python"})
diff --git a/azureai/model_client.py b/azureai/model_client.py
@@ -1,6 +1,6 @@
 import time
 
-from typing import Dict, Iterable, List, Optional, Union, Callable
+from typing import Dict, Iterable, List, Optional, Union, Callable, TypeVar
 from azure.ai.inference import ChatCompletionsClient, EmbeddingsClient
 from azure.ai.inference.models._models import (
     ChatRequestMessage,
@@ -15,6 +15,7 @@
 from statsig import statsig, StatsigUser, StatsigEvent
 from .get_statsig_user import get_statsig_user
 
+T = TypeVar( "T" )
 
 class InvokeContext:
     invoke_time: float
@@ -139,7 +140,7 @@ def task():
             return res
         return self.handle_errors(task=task, fallback=None)
 
-    def handle_errors[T](self, task: Callable[[], T], fallback: T) -> T:
+    def handle_errors(self, task: Callable[[], T], fallback: T) -> T:
         try:
             return task()
         except HttpResponseError as e: