Rename uses of rich.print to rprint

togethercomputer · Sep 24, 2024 · 7a157ca · 7a157ca
1 parent f323842
commit 7a157ca
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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,
             )