Fix lint

Author: orangetin

src/together/commands/chat.py

@@ -8,8 +8,10 @@
 from together import Complete
 from together.utils import get_logger
 
+
 logger = get_logger(str(__name__))
 
+
 def add_parser(subparsers: argparse._SubParsersAction[argparse.ArgumentParser]) -> None:
     COMMAND_NAME = "chat"
     subparser = subparsers.add_parser(COMMAND_NAME)

src/together/commands/files.py

@@ -164,7 +164,9 @@ def _run_check(args: argparse.Namespace) -> None:
 
 def _run_upload(args: argparse.Namespace) -> None:
     try:
-        response = Files.upload(file=args.file, check=not args.no_check, model=args.model)
+        response = Files.upload(
+            file=args.file, check=not args.no_check, model=args.model
+        )
     except together.AuthenticationError:
         logger.critical(together.MISSING_API_KEY_MESSAGE)
         exit(0)

src/together/commands/finetune.py

@@ -8,7 +8,8 @@
 
 import together
 from together import Finetune
-from together.utils import finetune_price_to_dollars, parse_timestamp, get_logger
+from together.utils import finetune_price_to_dollars, get_logger, parse_timestamp
+
 
 logger = get_logger(str(__name__))
 
@@ -352,7 +353,9 @@ def _run_list_events(args: argparse.Namespace) -> None:
 
 def _run_download(args: argparse.Namespace) -> None:
     try:
-        response = Finetune.download(args.fine_tune_id, args.output, args.checkpoint_step)
+        response = Finetune.download(
+            args.fine_tune_id, args.output, args.checkpoint_step
+        )
     except together.AuthenticationError:
         logger.critical(together.MISSING_API_KEY_MESSAGE)
         exit(0)

src/together/commands/models.py

@@ -8,6 +8,7 @@
 import together
 from together.utils import get_logger
 
+
 logger = get_logger(str(__name__))
 
 

src/together/error.py

@@ -74,9 +74,11 @@ def __init__(
             message, http_body, http_status, json_body, headers
         )
 
+
 class RateLimitError(TogetherException):
     pass
 
+
 class FileTypeError(TogetherException):
     pass
 

src/together/finetune.py

@@ -313,4 +313,3 @@ def download(
             session.close()
 
         return output  # this should be output file name
-

src/together/utils.py

@@ -1,7 +1,7 @@
 import logging
 import sys
 from datetime import datetime
-from typing import Any, Dict, Optional, Union
+from typing import Any, Dict, Optional
 
 import requests
 import sseclient  # type: ignore
@@ -57,6 +57,7 @@ def verify_api_key() -> None:
     if together.api_key is None:
         raise together.AuthenticationError(together.MISSING_API_KEY_MESSAGE)
 
+
 def extract_time(json_obj: Dict[str, Any]) -> int:
     try:
         return int(json_obj["created_at"])
@@ -98,7 +99,9 @@ def create_post_request(
         raise together.ResponseError(e)
 
     if response.status_code == 429:
-        raise together.RateLimitError(message="Too many requests received. Please pace your requests.")
+        raise together.RateLimitError(
+            message="Too many requests received. Please pace your requests."
+        )
     elif response.status_code == 500:
         raise Exception("Invalid API key supplied.")
     elif response.status_code == 401:
@@ -136,7 +139,9 @@ def create_get_request(
         raise together.ResponseError(e)
 
     if response.status_code == 429:
-        raise together.RateLimitError(message="Too many requests received. Please pace your requests.")
+        raise together.RateLimitError(
+            message="Too many requests received. Please pace your requests."
+        )
     elif response.status_code == 500:
         raise Exception("Invalid API key supplied.")
     elif response.status_code == 401: