From 6abc49c6117d5a169bdba8490520c44d8dcbc2b8 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 29 Jul 2024 20:01:12 -0700 Subject: [PATCH] fix linting --- litellm/types/llms/openai.py | 39 ++++++++++++++---------------------- 1 file changed, 15 insertions(+), 24 deletions(-) diff --git a/litellm/types/llms/openai.py b/litellm/types/llms/openai.py index fcff8b4babb..d6dbb8f5f14 100644 --- a/litellm/types/llms/openai.py +++ b/litellm/types/llms/openai.py @@ -458,15 +458,11 @@ class ChatCompletionUsageBlock(TypedDict): class Hyperparameters(TypedDict): - batch_size: Optional[Union[str, int]] = Field( - default="auto", description="Number of examples in each batch." - ) - learning_rate_multiplier: Optional[Union[str, float]] = Field( - default="auto", description="Scaling factor for the learning rate." - ) - n_epochs: Optional[Union[str, int]] = Field( - default="auto", description="The number of epochs to train the model for." - ) + batch_size: Optional[Union[str, int]] # "Number of examples in each batch." + learning_rate_multiplier: Optional[ + Union[str, float] + ] # Scaling factor for the learning rate + n_epochs: Optional[Union[str, int]] # "The number of epochs to train the model for" class FineTuningJobCreate(TypedDict): @@ -498,21 +494,16 @@ class FineTuningJobCreate(TypedDict): hyperparameters: Optional[Hyperparameters] = Field( default={}, description="The hyperparameters used for the fine-tuning job." ) - suffix: Optional[str] = Field( - default=None, - description="A string of up to 18 characters that will be added to your fine-tuned model name.", - ) - validation_file: Optional[str] = Field( - default=None, - description="The ID of an uploaded file that contains validation data.", - ) - integrations: Optional[List[str]] = Field( - default=None, - description="A list of integrations to enable for your fine-tuning job.", - ) - seed: Optional[int] = Field( - default=None, description="The seed controls the reproducibility of the job." - ) + suffix: Optional[ + str + ] # "A string of up to 18 characters that will be added to your fine-tuned model name." + validation_file: Optional[ + str + ] # "The ID of an uploaded file that contains validation data." + integrations: Optional[ + List[str] + ] # "A list of integrations to enable for your fine-tuning job." + seed: Optional[int] # "The seed controls the reproducibility of the job." class Config: allow_population_by_field_name = True