feat: remove kw_only=True from @DataClass for compatibility with Pyth…

…on versions before 3.10
ovuruska · Apr 29, 2024 · 9cade6f · 9cade6f
1 parent e0e56fa
commit 9cade6f
Show file tree

Hide file tree

Showing 10 changed files with 17 additions and 16 deletions.
diff --git a/deepinfra/types/automatic_speech_recognition/request.py b/deepinfra/types/automatic_speech_recognition/request.py
@@ -2,7 +2,7 @@
 from typing import Optional
 
 
-@dataclass(kw_only=True)
+@dataclass
 class AutomaticSpeechRecognitionRequest:
     audio: str
     task: Optional[str]

diff --git a/deepinfra/types/automatic_speech_recognition/response.py b/deepinfra/types/automatic_speech_recognition/response.py
@@ -5,15 +5,15 @@
 from deepinfra.types.common.inference_status import InferenceStatus
 
 
-@dataclass(kw_only=True)
+@dataclass
 class AutomaticSpeechRecognitionWord:
     text: str
     start: int
     end: int
     confidence: float
 
 
-@dataclass(kw_only=True)
+@dataclass
 class AutomaticSpeechRecognitionSegment:
     id: int
     seek: int
@@ -29,7 +29,7 @@ class AutomaticSpeechRecognitionSegment:
     words: List[AutomaticSpeechRecognitionWord]
 
 
-@dataclass(kw_only=True)
+@dataclass
 class AutomaticSpeechRecognitionResponse:
     text: str
     segments: List[AutomaticSpeechRecognitionSegment]

diff --git a/deepinfra/types/common/inference_status.py b/deepinfra/types/common/inference_status.py
@@ -1,7 +1,7 @@
 from dataclasses import dataclass
 
 
-@dataclass(kw_only=True)
+@dataclass
 class InferenceStatus:
     status: str
     runtime_ms: int
@@ -10,7 +10,7 @@ class InferenceStatus:
     tokens_input: int
 
 
-@dataclass(kw_only=True)
+@dataclass
 class Status:
     UNKNOWN = "unknown"
     QUEUED = "queued"

diff --git a/deepinfra/types/embeddings/request.py b/deepinfra/types/embeddings/request.py
@@ -2,7 +2,7 @@
 from typing import Optional, List
 
 
-@dataclass(kw_only=True)
+@dataclass
 class EmbeddingsRequest:
     inputs: List[str]
     normalize: Optional[bool] = None

diff --git a/deepinfra/types/embeddings/response.py b/deepinfra/types/embeddings/response.py
@@ -2,7 +2,7 @@
 from typing import Optional, List
 
 
-@dataclass(kw_only=True)
+@dataclass
 class EmbeddingStatus:
     status: str
     runtime_ms: int
@@ -11,7 +11,7 @@ class EmbeddingStatus:
     tokens_input: int
 
 
-@dataclass(kw_only=True)
+@dataclass
 class EmbeddingsResponse:
     embeddings: List[List[float]]
     input_tokens: int

diff --git a/deepinfra/types/image_generation/request.py b/deepinfra/types/image_generation/request.py
@@ -2,7 +2,7 @@
 from typing import Optional
 
 
-@dataclass(kw_only=True)
+@dataclass
 class ImageGenerationRequest:
     prompt: str
     negative_prompt: Optional[str] = None

diff --git a/deepinfra/types/image_generation/response.py b/deepinfra/types/image_generation/response.py
@@ -4,12 +4,12 @@
 from deepinfra.types.common.inference_status import InferenceStatus
 
 
-@dataclass(kw_only=True)
+@dataclass
 class Metrics:
     predict_time: int
 
 
-@dataclass(kw_only=True)
+@dataclass
 class ImageGenerationResponse:
     request_id: str
     inference_status: InferenceStatus

diff --git a/deepinfra/types/text_generation/request.py b/deepinfra/types/text_generation/request.py
@@ -2,7 +2,7 @@
 from typing import Optional, List, Dict
 
 
-@dataclass(kw_only=True)
+@dataclass
 class TextGenerationRequest:
     input: str
     stream: Optional[bool] = None

diff --git a/deepinfra/types/text_generation/response.py b/deepinfra/types/text_generation/response.py
@@ -4,12 +4,12 @@
 from deepinfra.types.common.inference_status import InferenceStatus
 
 
-@dataclass(kw_only=True)
+@dataclass
 class GeneratedText:
     generated_text: str
 
 
-@dataclass(kw_only=True)
+@dataclass
 class TextGenerationResponse:
     request_id: str
     inference_status: InferenceStatus

diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -1,2 +1,3 @@
 black==23.3.0
-mypy
+mypy
+types-requests