fix: subscriptable type

AI21Labs · Dec 12, 2023 · 71d1cc9 · 71d1cc9
1 parent cdaa0fd
commit 71d1cc9
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml
@@ -44,7 +44,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.8", "3.9", "3.10"]
+        python-version: ["3.7", "3.8", "3.9", "3.10", "3.11"]
     steps:
       - name: Checkout
         uses: actions/checkout@v3

diff --git a/ai21/resources/bases/improvements_base.py b/ai21/resources/bases/improvements_base.py
@@ -1,5 +1,5 @@
 from abc import ABC, abstractmethod
-from typing import Any, Dict
+from typing import Any, Dict, List
 
 from ai21.resources.responses.improvement_response import ImprovementsResponse
 
@@ -8,7 +8,7 @@ class Improvements(ABC):
     _module_name = "improvements"
 
     @abstractmethod
-    def create(self, text: str, types: list[str], **kwargs) -> ImprovementsResponse:
+    def create(self, text: str, types: List[str], **kwargs) -> ImprovementsResponse:
         pass
 
     def _json_to_response(self, json: Dict[str, Any]) -> ImprovementsResponse:

diff --git a/ai21/resources/responses/file_response.py b/ai21/resources/responses/file_response.py
@@ -1,5 +1,5 @@
 from datetime import date
-from typing import Optional
+from typing import Optional, List
 
 from ai21.models.ai21_base_model import AI21BaseModel
 
@@ -14,7 +14,7 @@ class FileResponse(AI21BaseModel):
     last_updated: date
     status: str
     path: Optional[str] = None
-    labels: Optional[list[str]] = None
+    labels: Optional[List[str]] = None
     public_url: Optional[str] = None
     error_code: Optional[int] = None
     error_message: Optional[str] = None