Fix typing

Distillative-AI · Aug 14, 2024 · bf8caa1 · bf8caa1
1 parent ba8b013
commit bf8caa1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dsp/modules/trainable_openai.py b/dsp/modules/trainable_openai.py
@@ -379,7 +379,7 @@ def load_from_job_id(job_id: str, **kwargs):
         model.fine_tuning_job_id = job_id
         return model
 
-    def get_finetune(self, train_path: str, val_path: str | None, method: TrainingMethod, **kwargs) -> Future[TrainableLM]:
+    def get_finetune(self, train_path: str, val_path: Optional[str], method: TrainingMethod, **kwargs) -> Future[TrainableLM]:
         """
         Does everything required to finetune an OpenAI model.