Linting

Author: martinRenou

src/sagemaker/local/entities.py

@@ -671,7 +671,8 @@ def start(self, **kwargs):
         self._executions[execution_id] = execution
         logger.info(
             "Starting execution for pipeline %s. Execution ID is %s",
-            self.pipeline.name, execution_id
+            self.pipeline.name,
+            execution_id,
         )
         self.last_modified_time = datetime.datetime.now().timestamp()
 
@@ -743,7 +744,9 @@ def update_execution_failure(self, step_name, failure_message):
         self.last_modified_time = datetime.datetime.now().timestamp()
         logger.info(
             "Pipeline execution %s FAILED because step "
-            "'%s' failed.", self.pipeline_execution_name, step_name
+            "'%s' failed.",
+            self.pipeline_execution_name,
+            step_name,
         )
 
     def update_step_properties(self, step_name, step_properties):

src/sagemaker/model_monitor/model_monitoring.py

@@ -676,7 +676,7 @@ def latest_monitoring_statistics(self, file_name=STATISTICS_JSON_DEFAULT_FILE_NA
         if len(executions) == 0:
             logger.warning(
                 "No executions found for schedule. monitoring_schedule_name: %s",
-                self.monitoring_schedule_name
+                self.monitoring_schedule_name,
             )
             return None
 
@@ -703,7 +703,7 @@ def latest_monitoring_constraint_violations(
         if len(executions) == 0:
             logger.warning(
                 "No executions found for schedule. monitoring_schedule_name: %s",
-                self.monitoring_schedule_name
+                self.monitoring_schedule_name,
             )
             return None
 
@@ -748,7 +748,7 @@ def list_executions(self):
         if len(monitoring_executions_dict["MonitoringExecutionSummaries"]) == 0:
             logger.warning(
                 "No executions found for schedule. monitoring_schedule_name: %s",
-                self.monitoring_schedule_name
+                self.monitoring_schedule_name,
             )
             return []
 
@@ -2435,7 +2435,7 @@ def latest_monitoring_statistics(self):
         if len(executions) == 0:
             logger.warning(
                 "No executions found for schedule. monitoring_schedule_name: %s",
-                self.monitoring_schedule_name
+                self.monitoring_schedule_name,
             )
             return None
 
@@ -2447,7 +2447,8 @@ def latest_monitoring_statistics(self):
             status = latest_monitoring_execution.describe()["ProcessingJobStatus"]
             logger.warning(
                 "Unable to retrieve statistics as job is in status '%s'. Latest statistics only "
-                "available for completed executions.", status
+                "available for completed executions.",
+                status,
             )
 
     def latest_monitoring_constraint_violations(self):
@@ -2464,7 +2465,7 @@ def latest_monitoring_constraint_violations(self):
         if len(executions) == 0:
             logger.warning(
                 "No executions found for schedule. monitoring_schedule_name: %s",
-                self.monitoring_schedule_name
+                self.monitoring_schedule_name,
             )
             return None
 
@@ -2475,7 +2476,8 @@ def latest_monitoring_constraint_violations(self):
             status = latest_monitoring_execution.describe()["ProcessingJobStatus"]
             logger.warning(
                 "Unable to retrieve constraint violations as job is in status '%s'. Latest "
-                "violations only available for completed executions.", status
+                "violations only available for completed executions.",
+                status,
             )
 
     @staticmethod

src/sagemaker/model_monitor/monitoring_files.py

@@ -130,7 +130,8 @@ def from_s3_uri(cls, statistics_file_s3_uri, kms_key=None, sagemaker_session=Non
                 "\nCould not retrieve statistics file at location '%s'. "
                 "To manually retrieve Statistics object from a given uri, "
                 "use 'my_model_monitor.statistics(my_s3_uri)' or "
-                "'Statistics.from_s3_uri(my_s3_uri)'", statistics_file_s3_uri
+                "'Statistics.from_s3_uri(my_s3_uri)'",
+                statistics_file_s3_uri,
             )
             raise error
 
@@ -260,7 +261,8 @@ def from_s3_uri(cls, constraints_file_s3_uri, kms_key=None, sagemaker_session=No
                 "\nCould not retrieve constraints file at location '%s'. "
                 "To manually retrieve Constraints object from a given uri, "
                 "use 'my_model_monitor.constraints(my_s3_uri)' or "
-                "'Constraints.from_s3_uri(my_s3_uri)'", constraints_file_s3_uri
+                "'Constraints.from_s3_uri(my_s3_uri)'",
+                constraints_file_s3_uri,
             )
             raise error
 

tests/integ/timeout.py

@@ -203,7 +203,8 @@ def _delete_schedules_associated_with_endpoint(sagemaker_session, endpoint_name)
         except Exception as e:
             logger.warning(
                 "Failed to delete monitor %s,\nError: %s",
-                monitor.monitoring_schedule_name, e
+                monitor.monitoring_schedule_name,
+                e,
             )