Skip to content

Commit

Permalink
delete the code that disables formatting for the tune function
Browse files Browse the repository at this point in the history
Signed-off-by: helenxie-bit <[email protected]>
  • Loading branch information
helenxie-bit committed Aug 30, 2024
1 parent e529ec4 commit 17f9dea
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions sdk/python/v1beta1/kubeflow/katib/api/katib_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -165,14 +165,16 @@ def create_experiment(
)
)

# fmt: off
def tune(
self,
# TODO (andreyvelich): How to be consistent with other APIs (name) ?
name: str,
model_provider_parameters: Optional["HuggingFaceModelParams"] = None, # noqa: F821
dataset_provider_parameters: Optional[Union[
"HuggingFaceDatasetParams", "S3DatasetParams"]] = None, # noqa: F821
model_provider_parameters: Optional[
"HuggingFaceModelParams" # noqa: F821
] = None,
dataset_provider_parameters: Optional[
Union["HuggingFaceDatasetParams", "S3DatasetParams"] # noqa: F821
] = None,
trainer_parameters: Optional["HuggingFaceTrainerParams"] = None, # noqa: F821
storage_config: Optional[Dict[str, Optional[Union[str, List[str]]]]] = {
"size": constants.PVC_DEFAULT_SIZE,
Expand Down Expand Up @@ -205,8 +207,6 @@ def tune(
pip_index_url: str = "https://pypi.org/simple",
metrics_collector_config: Dict[str, Any] = {"kind": "StdOut"},
):
# fmt: on

"""
Create HyperParameter Tuning Katib Experiment using one of the following
options:
Expand Down

0 comments on commit 17f9dea

Please sign in to comment.