Allow fine tune hyperparams non-string values (#136)

Author: seanshi-scale

launch/api_client/model/create_fine_tune_request.py

@@ -40,7 +40,74 @@ class MetaOapg:
  class properties:
  class hyperparameters(schemas.DictSchema):
  class MetaOapg:
- additional_properties = schemas.StrSchema
+ class additional_properties(
+ schemas.ComposedSchema,
+ ):
+ class MetaOapg:
+ any_of_0 = schemas.StrSchema
+ any_of_1 = schemas.IntSchema
+ any_of_2 = schemas.NumberSchema
+
+ @classmethod
+ @functools.lru_cache()
+ def any_of(cls):
+ # we need this here to make our import statements work
+ # we must store _composed_schemas in here so the code is only run
+ # when we invoke this method. If we kept this at the class
+ # level we would get an error because the class level
+ # code would be run when this module is imported, and these composed
+ # classes don't exist yet because their module has not finished
+ # loading
+ return [
+ cls.any_of_0,
+ cls.any_of_1,
+ cls.any_of_2,
+ ]
+
+ def __new__(
+ cls,
+ *_args: typing.Union[
+ dict,
+ frozendict.frozendict,
+ str,
+ date,
+ datetime,
+ uuid.UUID,
+ int,
+ float,
+ decimal.Decimal,
+ bool,
+ None,
+ list,
+ tuple,
+ bytes,
+ io.FileIO,
+ io.BufferedReader,
+ ],
+ _configuration: typing.Optional[schemas.Configuration] = None,
+ **kwargs: typing.Union[
+ schemas.AnyTypeSchema,
+ dict,
+ frozendict.frozendict,
+ str,
+ date,
+ datetime,
+ uuid.UUID,
+ int,
+ float,
+ decimal.Decimal,
+ None,
+ list,
+ tuple,
+ bytes,
+ ],
+ ) -> "additional_properties":
+ return super().__new__(
+ cls,
+ *_args,
+ _configuration=_configuration,
+ **kwargs,
+ )
 
  def __getitem__(
  self,
@@ -64,7 +131,22 @@ def __new__(
  _configuration: typing.Optional[schemas.Configuration] = None,
  **kwargs: typing.Union[
  MetaOapg.additional_properties,
+ dict,
+ frozendict.frozendict,
  str,
+ date,
+ datetime,
+ uuid.UUID,
+ int,
+ float,
+ decimal.Decimal,
+ bool,
+ None,
+ list,
+ tuple,
+ bytes,
+ io.FileIO,
+ io.BufferedReader,
  ],
  ) -> "hyperparameters":
  return super().__new__(

launch/api_client/model/create_fine_tune_request.pyi

@@ -38,7 +38,73 @@ class CreateFineTuneRequest(schemas.DictSchema):
  class properties:
  class hyperparameters(schemas.DictSchema):
  class MetaOapg:
- additional_properties = schemas.StrSchema
+ class additional_properties(
+ schemas.ComposedSchema,
+ ):
+ class MetaOapg:
+ any_of_0 = schemas.StrSchema
+ any_of_1 = schemas.IntSchema
+ any_of_2 = schemas.NumberSchema
+
+ @classmethod
+ @functools.lru_cache()
+ def any_of(cls):
+ # we need this here to make our import statements work
+ # we must store _composed_schemas in here so the code is only run
+ # when we invoke this method. If we kept this at the class
+ # level we would get an error because the class level
+ # code would be run when this module is imported, and these composed
+ # classes don't exist yet because their module has not finished
+ # loading
+ return [
+ cls.any_of_0,
+ cls.any_of_1,
+ cls.any_of_2,
+ ]
+ def __new__(
+ cls,
+ *_args: typing.Union[
+ dict,
+ frozendict.frozendict,
+ str,
+ date,
+ datetime,
+ uuid.UUID,
+ int,
+ float,
+ decimal.Decimal,
+ bool,
+ None,
+ list,
+ tuple,
+ bytes,
+ io.FileIO,
+ io.BufferedReader,
+ ],
+ _configuration: typing.Optional[schemas.Configuration] = None,
+ **kwargs: typing.Union[
+ schemas.AnyTypeSchema,
+ dict,
+ frozendict.frozendict,
+ str,
+ date,
+ datetime,
+ uuid.UUID,
+ int,
+ float,
+ decimal.Decimal,
+ None,
+ list,
+ tuple,
+ bytes,
+ ],
+ ) -> "additional_properties":
+ return super().__new__(
+ cls,
+ *_args,
+ _configuration=_configuration,
+ **kwargs,
+ )
  def __getitem__(
  self,
  name: typing.Union[str,],
@@ -59,7 +125,22 @@ class CreateFineTuneRequest(schemas.DictSchema):
  _configuration: typing.Optional[schemas.Configuration] = None,
  **kwargs: typing.Union[
  MetaOapg.additional_properties,
+ dict,
+ frozendict.frozendict,
  str,
+ date,
+ datetime,
+ uuid.UUID,
+ int,
+ float,
+ decimal.Decimal,
+ bool,
+ None,
+ list,
+ tuple,
+ bytes,
+ io.FileIO,
+ io.BufferedReader,
  ],
  ) -> "hyperparameters":
  return super().__new__(