Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Aug 15, 2024
1 parent 8a7061b commit da99a10
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions comps/finetuning/handlers.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,13 +9,11 @@

from envs import CHECK_JOB_STATUS_INTERVAL, DATASET_BASE_PATH, MODEL_CONFIG_FILE_MAP, ray_client
from finetune_config import FinetuneConfig

from pydantic_yaml import parse_yaml_raw_as, to_yaml_file
from ray.job_submission import JobSubmissionClient

from comps.cores.proto.api_protocol import FineTuningJob, FineTuningJobsRequest

from ray.job_submission import JobSubmissionClient

FineTuningJobID = str
running_finetuning_jobs: Dict[FineTuningJobID, FineTuningJob] = {}
finetuning_job_to_ray_job: Dict[FineTuningJobID, str] = {}
Expand All @@ -38,7 +36,7 @@ def handle_create_finetuning_jobs(request: FineTuningJobsRequest):
finetune_config = parse_yaml_raw_as(FinetuneConfig, f)

finetune_config.Dataset.train_file = train_file_path

if request.hyperparameters is not None:
if request.hyperparameters.epochs != "auto":
finetune_config.Training.epochs = request.hyperparameters.epochs
Expand Down

0 comments on commit da99a10

Please sign in to comment.