Tabulate view for models info and remove commented out code

Author: orangetin

src/together/commands/finetune.py

@@ -342,9 +342,3 @@ def _run_checkpoint(args: argparse.Namespace) -> None:
     checkpoints = Finetune.get_checkpoints(args.fine_tune_id)
     print(json.dumps(checkpoints, indent=4))
     print(f"\n{len(checkpoints)} checkpoints found")
-
-
-# def _run_delete_model(args: argparse.Namespace) -> None:
-#     finetune = Finetune(args.endpoint)
-#     response = finetune.delete_finetune_model(args.model)
-#     print(json.dumps(response))

src/together/commands/models.py

@@ -147,7 +147,11 @@ def _run_list(args: argparse.Namespace) -> None:
 
 
 def _run_info(args: argparse.Namespace) -> None:
-    if not args.raw:
+    model_info = together.Models.info(args.model)
+
+    if args.raw:
+        print(json.dumps(model_info, indent=4))
+    else:
         hidden_keys = [
             "_id",
             "modelInstanceConfig",
@@ -159,13 +163,16 @@ def _run_info(args: argparse.Namespace) -> None:
             "pricing_tier",
             "hardware_label",
             "depth",
+            "descriptionLink",
         ]
-    else:
-        hidden_keys = []
 
-    model_info = together.Models.info(args.model, hidden_keys=hidden_keys)
-
-    print(json.dumps(model_info, indent=4))
+        table_data = [
+            {"Key": key, "Value": value}
+            for key, value in model_info.items()
+            if key not in hidden_keys
+        ]
+        table = tabulate(table_data, tablefmt="grid")
+        print(table)
 
 
 def _run_instances(args: argparse.Namespace) -> None:

src/together/files.py

@@ -133,10 +133,6 @@ def upload(
             logger.critical(f"Response error raised: {e}")
             raise together.ResponseError(e)
 
-        # output_dict["filename"] = os.path.basename(file)
-        # output_dict["id"] = str(file_id)
-        # output_dict["object"] = "file"
-
         return {
             "filename": os.path.basename(file),
             "id": str(file_id),