fix lint

clarifai/runners/models/model_upload.py

@@ -1,12 +1,10 @@
-import json
 import os
 import re
 import sys
 import tarfile
 import time
 from string import Template
 
-import requests
 import yaml
 from clarifai_grpc.grpc.api import resources_pb2, service_pb2
 from clarifai_grpc.grpc.api.status import status_code_pb2
@@ -22,6 +20,7 @@
 from clarifai.urls.helper import ClarifaiUrlHelper
 from clarifai.utils.logging import logger
 
+
 def _clear_line(n: int = 1) -> None:
   LINE_UP = '\033[1A'  # Move cursor up one line
   LINE_CLEAR = '\x1b[2K'  # Clear the entire line
@@ -30,7 +29,7 @@ def _clear_line(n: int = 1) -> None:
 
 
 class ModelUploader:
-  DEFAULT_CHECKPOINT_SIZE = 50*1024**3 # 50 GiB
+  DEFAULT_CHECKPOINT_SIZE = 50 * 1024**3  # 50 GiB
 
   def __init__(self, folder: str, validate_api_ids: bool = True, download_validation_only=False):
     """
@@ -597,6 +596,7 @@ def monitor_model_build(self):
             f"\nModel build failed with status: {resp.model_version.status} and response {resp}")
         return False
 
+
 def main(folder, download_checkpoints, skip_dockerfile):
   uploader = ModelUploader(folder)
   if download_checkpoints:

clarifai/runners/utils/loader.py

@@ -2,10 +2,11 @@
 import importlib.util
 import json
 import os
-import requests
 import shutil
 import subprocess
 
+import requests
+
 from clarifai.utils.logging import logger
 
 
@@ -196,6 +197,5 @@ def get_huggingface_checkpoint_total_size(repo_name):
         total_size += file['size']
       return total_size
     except Exception as e:
-      logger.error(
-          f"Error fetching checkpoint size from huggingface.co: {e}")
+      logger.error(f"Error fetching checkpoint size from huggingface.co: {e}")
       return 0