ENG-10038: Use Union instead of Conditional Or For DownloadCheckpoint…

…TypeChoice Typing (#177) * Use Union instead of | * Add default case for training type as full * Swap from union to future annotations --------- Co-authored-by: Arsh Zahed <[email protected]>
togethercomputer · Aug 28, 2024 · 469b723 · 469b723
1 parent 5f504ed
commit 469b723
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/together/cli/api/finetune.py b/src/together/cli/api/finetune.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import json
 from textwrap import wrap
 

diff --git a/src/together/types/finetune.py b/src/together/types/finetune.py
@@ -230,7 +230,7 @@ class FinetuneResponse(BaseModel):
     @field_validator("training_type")
     @classmethod
     def validate_training_type(cls, v: TrainingType) -> TrainingType:
-        if v.type == "Full":
+        if v.type == "Full" or v.type == "":
             return FullTrainingType(**v.model_dump())
         elif v.type == "Lora":
             return LoRATrainingType(**v.model_dump())