Skip to content

Commit e8e731c

Browse files
VipulMascarenhasmayoor
authored andcommitted
Implement log to watch logs when log id is provided
1 parent 72c3d26 commit e8e731c

File tree

1 file changed

+23
-0
lines changed

1 file changed

+23
-0
lines changed

ads/aqua/finetuning/finetuning.py

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44

55
import json
66
import os
7+
import time
78
from typing import Dict
89

910
from oci.data_science.models import (
@@ -416,6 +417,28 @@ def create(
416417
value=source.display_name,
417418
)
418419

420+
if create_fine_tuning_details.watch_logs:
421+
if (
422+
create_fine_tuning_details.log_id
423+
and create_fine_tuning_details.log_group_id
424+
):
425+
logger.info(
426+
f"Watching fine-tuning job run logs for {ft_job_run.id}. Press Ctrl+C stop watching logs.\n"
427+
)
428+
try:
429+
ft_job_run.watch()
430+
except KeyboardInterrupt:
431+
logger.info(f"\nStopped watching logs for {ft_job_run.id}.\n")
432+
time.sleep(2)
433+
except Exception as ex:
434+
logger.debug(
435+
f"Something unexpected occurred while watching logs.\n{str(ex)}"
436+
)
437+
else:
438+
logger.info(
439+
"Logging details are not provided, set `log_id` and `log_group_id` to watch logs."
440+
)
441+
419442
return AquaFineTuningSummary(
420443
id=ft_model.id,
421444
name=ft_model.display_name,

0 commit comments

Comments
 (0)