Wandb update (#150)

* Wandb update * update

Author: yunfeng-scale

launch/api_client/model/create_fine_tune_request.py

@@ -160,12 +160,14 @@ def __new__(
  training_file = schemas.StrSchema
  suffix = schemas.StrSchema
  validation_file = schemas.StrSchema
+ wandb_config = schemas.DictSchema
  __annotations__ = {
  "hyperparameters": hyperparameters,
  "model": model,
  "training_file": training_file,
  "suffix": suffix,
  "validation_file": validation_file,
+ "wandb_config": wandb_config,
  }
 
  training_file: MetaOapg.properties.training_file
@@ -192,6 +194,10 @@ def __getitem__(self, name: typing_extensions.Literal["suffix"]) -> MetaOapg.pro
  def __getitem__(self, name: typing_extensions.Literal["validation_file"]) -> MetaOapg.properties.validation_file:
  ...
 
+ @typing.overload
+ def __getitem__(self, name: typing_extensions.Literal["wandb_config"]) -> MetaOapg.properties.wandb_config:
+ ...
+
  @typing.overload
  def __getitem__(self, name: str) -> schemas.UnsetAnyTypeSchema:
  ...
@@ -205,6 +211,7 @@ def __getitem__(
  "training_file",
  "suffix",
  "validation_file",
+ "wandb_config",
  ],
  str,
  ],
@@ -236,6 +243,12 @@ def get_item_oapg(
  ) -> typing.Union[MetaOapg.properties.validation_file, schemas.Unset]:
  ...
 
+ @typing.overload
+ def get_item_oapg(
+ self, name: typing_extensions.Literal["wandb_config"]
+ ) -> typing.Union[MetaOapg.properties.wandb_config, schemas.Unset]:
+ ...
+
  @typing.overload
  def get_item_oapg(self, name: str) -> typing.Union[schemas.UnsetAnyTypeSchema, schemas.Unset]:
  ...
@@ -249,6 +262,7 @@ def get_item_oapg(
  "training_file",
  "suffix",
  "validation_file",
+ "wandb_config",
  ],
  str,
  ],
@@ -276,6 +290,9 @@ def __new__(
  ],
  suffix: typing.Union[MetaOapg.properties.suffix, str, schemas.Unset] = schemas.unset,
  validation_file: typing.Union[MetaOapg.properties.validation_file, str, schemas.Unset] = schemas.unset,
+ wandb_config: typing.Union[
+ MetaOapg.properties.wandb_config, dict, frozendict.frozendict, schemas.Unset
+ ] = schemas.unset,
  _configuration: typing.Optional[schemas.Configuration] = None,
  **kwargs: typing.Union[
  schemas.AnyTypeSchema,
@@ -302,6 +319,7 @@ def __new__(
  model=model,
  suffix=suffix,
  validation_file=validation_file,
+ wandb_config=wandb_config,
  _configuration=_configuration,
  **kwargs,
  )

launch/api_client/model/create_fine_tune_request.pyi

@@ -153,12 +153,14 @@ class CreateFineTuneRequest(schemas.DictSchema):
  training_file = schemas.StrSchema
  suffix = schemas.StrSchema
  validation_file = schemas.StrSchema
+ wandb_config = schemas.DictSchema
  __annotations__ = {
  "hyperparameters": hyperparameters,
  "model": model,
  "training_file": training_file,
  "suffix": suffix,
  "validation_file": validation_file,
+ "wandb_config": wandb_config,
  }
  training_file: MetaOapg.properties.training_file
  hyperparameters: MetaOapg.properties.hyperparameters
@@ -179,6 +181,8 @@ class CreateFineTuneRequest(schemas.DictSchema):
  self, name: typing_extensions.Literal["validation_file"]
  ) -> MetaOapg.properties.validation_file: ...
  @typing.overload
+ def __getitem__(self, name: typing_extensions.Literal["wandb_config"]) -> MetaOapg.properties.wandb_config: ...
+ @typing.overload
  def __getitem__(self, name: str) -> schemas.UnsetAnyTypeSchema: ...
  def __getitem__(
  self,
@@ -189,6 +193,7 @@ class CreateFineTuneRequest(schemas.DictSchema):
  "training_file",
  "suffix",
  "validation_file",
+ "wandb_config",
  ],
  str,
  ],
@@ -212,6 +217,10 @@ class CreateFineTuneRequest(schemas.DictSchema):
  self, name: typing_extensions.Literal["validation_file"]
  ) -> typing.Union[MetaOapg.properties.validation_file, schemas.Unset]: ...
  @typing.overload
+ def get_item_oapg(
+ self, name: typing_extensions.Literal["wandb_config"]
+ ) -> typing.Union[MetaOapg.properties.wandb_config, schemas.Unset]: ...
+ @typing.overload
  def get_item_oapg(self, name: str) -> typing.Union[schemas.UnsetAnyTypeSchema, schemas.Unset]: ...
  def get_item_oapg(
  self,
@@ -222,6 +231,7 @@ class CreateFineTuneRequest(schemas.DictSchema):
  "training_file",
  "suffix",
  "validation_file",
+ "wandb_config",
  ],
  str,
  ],
@@ -248,6 +258,9 @@ class CreateFineTuneRequest(schemas.DictSchema):
  ],
  suffix: typing.Union[MetaOapg.properties.suffix, str, schemas.Unset] = schemas.unset,
  validation_file: typing.Union[MetaOapg.properties.validation_file, str, schemas.Unset] = schemas.unset,
+ wandb_config: typing.Union[
+ MetaOapg.properties.wandb_config, dict, frozendict.frozendict, schemas.Unset
+ ] = schemas.unset,
  _configuration: typing.Optional[schemas.Configuration] = None,
  **kwargs: typing.Union[
  schemas.AnyTypeSchema,
@@ -274,6 +287,7 @@ class CreateFineTuneRequest(schemas.DictSchema):
  model=model,
  suffix=suffix,
  validation_file=validation_file,
+ wandb_config=wandb_config,
  _configuration=_configuration,
  **kwargs,
  )

launch/client.py

@@ -2991,6 +2991,7 @@ def create_fine_tune(
  validation_file: Optional[str] = None,
  fine_tuning_method: Optional[str] = None,
  hyperparameters: Optional[Dict[str, str]] = None,
+ wandb_config: Optional[Dict[str, Any]] = None,
  suffix: str = None,
  ) -> CreateFineTuneResponse:
  """
@@ -3006,6 +3007,7 @@ def create_fine_tune(
  fine_tuning_method: Fine-tuning method. Currently unused,
  but when different techniques are implemented we will expose this field.
  hyperparameters: Hyperparameters to pass in to training job.
+ wandb_config: Configuration for Weights and Biases. `api_key` must be provided which is the API key.
  suffix: Optional user-provided identifier suffix for the fine-tuned model.
 
  Returns:
@@ -3020,6 +3022,7 @@ def create_fine_tune(
  validation_file=validation_file,
  fine_tuning_method=fine_tuning_method,
  hyperparameters=hyperparameters,
+ wandb_config=wandb_config,
  suffix=suffix,
  )
  )