diff --git a/src/together/cli/api/finetune.py b/src/together/cli/api/finetune.py index 1a9a8c07..548545ea 100644 --- a/src/together/cli/api/finetune.py +++ b/src/together/cli/api/finetune.py @@ -6,7 +6,7 @@ import click from click.core import ParameterSource # type: ignore[attr-defined] -from rich import print as rich_print +from rich import print as rprint from tabulate import tabulate from together import Together @@ -136,7 +136,7 @@ def create( # created_at reports UTC time, we use .astimezone() to convert to local time formatted_time = created_time.astimezone().strftime("%m/%d/%Y, %H:%M:%S") report_string += f" at {formatted_time}" - rich_print(report_string) + rprint(report_string) @fine_tuning.command() diff --git a/src/together/resources/finetune.py b/src/together/resources/finetune.py index 47f0b3a2..8a09d421 100644 --- a/src/together/resources/finetune.py +++ b/src/together/resources/finetune.py @@ -2,7 +2,7 @@ from pathlib import Path -from rich import print +from rich import print as rprint from together.abstract import api_requestor from together.filemanager import DownloadManager @@ -104,7 +104,7 @@ def create( wandb_key=wandb_api_key, ) if verbose: - print( + rprint( "Submitting a fine-tuning job with the following parameters:", finetune_request, )