formatted files

Author: ananth102

src/sagemaker/experiments/run.py

@@ -205,7 +205,9 @@ def __init__(
             )
 
         if not _TrialComponent._trial_component_is_associated_to_trial(
-            self._trial_component.trial_component_name, self._trial.trial_name, sagemaker_session
+            self._trial_component.trial_component_name,
+            self._trial.trial_name,
+            sagemaker_session,
         ):
             self._trial.add_trial_component(self._trial_component)
 
@@ -340,7 +342,9 @@ def log_precision_recall(
         if positive_label is not None:
             kwargs["pos_label"] = positive_label
 
-        precision, recall, _ = precision_recall_curve(y_true, predicted_probabilities, **kwargs)
+        precision, recall, _ = precision_recall_curve(
+            y_true, predicted_probabilities, **kwargs
+        )
 
         kwargs["average"] = "micro"
         ap = average_precision_score(y_true, predicted_probabilities, **kwargs)
@@ -560,7 +564,9 @@ def _is_input_valid(input_type, field_name, field_value) -> bool:
             field_name (str): The name of the field to be checked.
             field_value (str or int or float): The value of the field to be checked.
         """
-        if isinstance(field_value, Number) and (isnan(field_value) or isinf(field_value)):
+        if isinstance(field_value, Number) and (
+            isnan(field_value) or isinf(field_value)
+        ):
             logger.warning(
                 "Failed to log %s %s. Received invalid value: %s.",
                 input_type,
@@ -622,10 +628,14 @@ def _verify_trial_component_artifacts_length(self, is_output):
         err_msg_template = "Cannot add more than {} {}_artifacts under run"
         if is_output:
             if len(self._trial_component.output_artifacts) >= MAX_RUN_TC_ARTIFACTS_LEN:
-                raise ValueError(err_msg_template.format(MAX_RUN_TC_ARTIFACTS_LEN, "output"))
+                raise ValueError(
+                    err_msg_template.format(MAX_RUN_TC_ARTIFACTS_LEN, "output")
+                )
         else:
             if len(self._trial_component.input_artifacts) >= MAX_RUN_TC_ARTIFACTS_LEN:
-                raise ValueError(err_msg_template.format(MAX_RUN_TC_ARTIFACTS_LEN, "input"))
+                raise ValueError(
+                    err_msg_template.format(MAX_RUN_TC_ARTIFACTS_LEN, "input")
+                )
 
     @staticmethod
     def _generate_trial_component_name(run_name: str, experiment_name: str) -> str:
@@ -646,20 +656,28 @@ def _generate_trial_component_name(run_name: str, experiment_name: str) -> str:
         """
         buffer = 1  # leave length buffers for delimiters
         max_len = int(MAX_NAME_LEN_IN_BACKEND / 2) - buffer
-        err_msg_template = "The {} (length: {}) must have length less than or equal to {}"
+        err_msg_template = (
+            "The {} (length: {}) must have length less than or equal to {}"
+        )
         if len(run_name) > max_len:
-            raise ValueError(err_msg_template.format("run_name", len(run_name), max_len))
+            raise ValueError(
+                err_msg_template.format("run_name", len(run_name), max_len)
+            )
         if len(experiment_name) > max_len:
             raise ValueError(
-                err_msg_template.format("experiment_name", len(experiment_name), max_len)
+                err_msg_template.format(
+                    "experiment_name", len(experiment_name), max_len
+                )
             )
         trial_component_name = "{}{}{}".format(experiment_name, DELIMITER, run_name)
         # due to mixed-case concerns on the backend
         trial_component_name = trial_component_name.lower()
         return trial_component_name
 
     @staticmethod
-    def _extract_run_name_from_tc_name(trial_component_name: str, experiment_name: str) -> str:
+    def _extract_run_name_from_tc_name(
+        trial_component_name: str, experiment_name: str
+    ) -> str:
         """Extract the user supplied run name from a trial component name.
 
         Args:
@@ -676,7 +694,9 @@ def _extract_run_name_from_tc_name(trial_component_name: str, experiment_name: s
         )
 
     @staticmethod
-    def _append_run_tc_label_to_tags(tags: Optional[List[Dict[str, str]]] = None) -> list:
+    def _append_run_tc_label_to_tags(
+        tags: Optional[List[Dict[str, str]]] = None
+    ) -> list:
         """Append the run trial component label to tags used to create a trial component.
 
         Args: