ci is fixed

ovuruska · Apr 26, 2024 · 35fe6cd · 35fe6cd
1 parent ee91c7f
commit 35fe6cd
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 9 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -15,7 +15,7 @@ jobs:
 
     strategy:
       matrix:
-        python-version: [3.7, 3.9, 3.12]
+        python-version: [3.7, 3.9, 3.11]
 
     steps:
       - name: Checkout code
@@ -34,4 +34,4 @@ jobs:
 
       - name: Run lint check
         run: |
-          black --check --verbose deepinfra
+          black --check --verbose  deepinfra
diff --git a/deepinfra/models/base/automatic_speech_recognition.py b/deepinfra/models/base/automatic_speech_recognition.py
@@ -1,7 +1,4 @@
 from deepinfra.models.base import BaseModel
-from deepinfra.types.automatic_speech_recognition.request import (
-    AutomaticSpeechRecognitionRequest,
-)
 from deepinfra.types.automatic_speech_recognition.response import (
     AutomaticSpeechRecognitionResponse,
 )
@@ -20,9 +17,7 @@ class AutomaticSpeechRecognition(BaseModel):
     def __init__(self, endpoint: str, auth_token: str = None):
         super().__init__(endpoint, auth_token)
 
-    def generate(
-        self, body: AutomaticSpeechRecognitionRequest
-    ) -> AutomaticSpeechRecognitionResponse:
+    def generate(self, body) -> AutomaticSpeechRecognitionResponse:
         """
         Generates the automatic speech recognition response.
         @param body: The request body.

diff --git a/deepinfra/models/base/image_generation.py b/deepinfra/models/base/image_generation.py
@@ -1,4 +1,4 @@
-from deepinfra.models.base.base import BaseModel
+from .base import BaseModel
 
 
 class ImageGeneration(BaseModel):