Skip to content

Commit

Permalink
formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
greenw0lf committed Oct 4, 2024
1 parent 976b51f commit 2fa972a
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion model_download.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,9 @@
from config import model_base_dir, w_model, s3_endpoint_url


def extract_model(w_model: str, model_base_dir: str, destination: str, asset_id: str, extension: str) -> str:
def extract_model(
w_model: str, model_base_dir: str, destination: str, asset_id: str, extension: str
) -> str:
logger = logging.getLogger(__name__)
logger.info(f"Downloaded {w_model} into {model_base_dir}")
if not os.path.exists(destination): # Create dir for model to be extracted in
Expand Down

0 comments on commit 2fa972a

Please sign in to comment.