lint

Author: bencrabtree

src/sagemaker/jumpstart/curated_hub/accessors/filegenerator.py

@@ -61,6 +61,7 @@ def format(self, file_input: S3ObjectLocation) -> List[FileInfo]:
             files.append(FileInfo(key, size, last_modified))
         return files
 
+
 class ModelSpecsFileGenerator(FileGenerator):
     """Utility class to help format all data paths from JumpStart public model specs."""
 

src/sagemaker/jumpstart/curated_hub/accessors/public_model_data.py

@@ -84,13 +84,14 @@ def demo_notebook_s3_reference(self):
         framework = get_model_framework(self.model_specs)
         key = f"{framework}-notebooks/{self.model_specs.model_id}-inference.ipynb"
         return S3ObjectLocation(self._get_bucket_name(), key)
-    
+
     @property
     def markdown_s3_reference(self):
         """Retrieves s3 reference for model markdown"""
         framework = get_model_framework(self.model_specs)
         key = f"{framework}-metadata/{self.model_specs.model_id}.md"
         return S3ObjectLocation(self._get_bucket_name(), key)
+
     def _get_bucket_name(self) -> str:
         """Retrieves s3 bucket"""
         return self._bucket

src/sagemaker/jumpstart/curated_hub/curated_hub.py

@@ -20,7 +20,10 @@
 from botocore.client import BaseClient
 from packaging.version import Version
 
-from sagemaker.jumpstart.curated_hub.accessors.filegenerator import FileGenerator, ModelSpecsFileGenerator, S3PathFileGenerator
+from sagemaker.jumpstart.curated_hub.accessors.filegenerator import (
+    ModelSpecsFileGenerator,
+    S3PathFileGenerator,
+)
 from sagemaker.jumpstart.curated_hub.accessors.objectlocation import S3ObjectLocation
 from sagemaker.jumpstart.curated_hub.accessors.sync import FileSync
 from sagemaker.jumpstart.enums import JumpStartScriptScope
@@ -296,7 +299,9 @@ def _sync_public_model_to_hub(self, model: Dict[str, str]):
         )
         # TODO: Validations? HeadBucket?
 
-        src_files = ModelSpecsFileGenerator(self.region, self._s3_client, studio_specs).format(model_specs)
+        src_files = ModelSpecsFileGenerator(self.region, self._s3_client, studio_specs).format(
+            model_specs
+        )
         dest_files = S3PathFileGenerator(self.region, self._s3_client).format(dest_location)
 
         files_to_copy = FileSync(src_files, dest_files, dest_location).call()

tests/unit/sagemaker/jumpstart/curated_hub/test_filegenerator.py

@@ -13,7 +13,10 @@
 from __future__ import absolute_import
 import pytest
 from unittest.mock import Mock, patch
-from sagemaker.jumpstart.curated_hub.accessors.filegenerator import FileGenerator, ModelSpecsFileGenerator, S3PathFileGenerator
+from sagemaker.jumpstart.curated_hub.accessors.filegenerator import (
+    ModelSpecsFileGenerator,
+    S3PathFileGenerator,
+)
 from sagemaker.jumpstart.curated_hub.accessors.fileinfo import FileInfo
 
 from sagemaker.jumpstart.curated_hub.accessors.objectlocation import S3ObjectLocation