Fix formatting

Author: navinns

src/sagemaker/inputs.py

@@ -143,7 +143,6 @@ class CompilationInput(object):
     It can be used when calling ``sagemaker.model.Model.compile_model()``
 
     Parameters:
-
         target_instance_type(str): Identifies the device that you want to
                 run your model after compilation, for example: ml_c5. For allowed
                 strings see
@@ -192,6 +191,7 @@ class CompilationInput(object):
     target_platform_accelerator: str = attr.ib(default=None)
     compiler_options: dict = attr.ib(factory=dict)
 
+
 @attr.s
 class TransformInput(object):
     """Create a class containing all the parameters.

src/sagemaker/model.py

@@ -412,12 +412,13 @@ def _compilation_job_config(
         }
 
     def _get_compilation_args(self, estimator, inputs):
-        """Constructs a dict of arguments for an Amazon SageMaker compilation job from the estimator.
-                Args:
-                    estimator (sagemaker.estimator.EstimatorBase): Estimator object
-                        created by the user.
-                    inputs (CompilationInput): class containing all the parameters that
-                        can be used when calling ``sagemaker.model.Model.compile_model()``
+        """Constructs a dict of arguments for an Amazon SageMaker compilation job from estimator.
+
+        Args:
+            estimator (sagemaker.estimator.EstimatorBase): Estimator object
+                created by the user.
+            inputs (CompilationInput): class containing all the parameters that
+                can be used when calling ``sagemaker.model.Model.compile_model()``
         """
 
         if not isinstance(inputs, CompilationInput):

src/sagemaker/session.py

@@ -1848,7 +1848,7 @@ def _get_compilation_request(
         role,
         stop_condition,
         tags=None,
-        vpc_config=None
+        vpc_config=None,
     ):
         """Construct CreateCompilationJob request
 

src/sagemaker/workflow/steps.py

@@ -14,14 +14,19 @@
 from __future__ import absolute_import
 
 import abc
-
 from enum import Enum
 from typing import Dict, List, Union
 
 import attr
 
 from sagemaker.estimator import EstimatorBase, _TrainingJob
-from sagemaker.inputs import CreateModelInput, TrainingInput, TransformInput, FileSystemInput, CompilationInput
+from sagemaker.inputs import (
+    CompilationInput,
+    CreateModelInput,
+    FileSystemInput,
+    TrainingInput,
+    TransformInput,
+)
 from sagemaker.model import Model
 from sagemaker.processing import (
     ProcessingInput,
@@ -31,16 +36,9 @@
 )
 from sagemaker.transformer import Transformer, _TransformJob
 from sagemaker.tuner import HyperparameterTuner, _TuningJob
-from sagemaker.workflow.entities import (
-    DefaultEnumMeta,
-    Entity,
-    RequestType,
-)
-from sagemaker.workflow.properties import (
-    PropertyFile,
-    Properties,
-)
+from sagemaker.workflow.entities import DefaultEnumMeta, Entity, RequestType
 from sagemaker.workflow.functions import Join
+from sagemaker.workflow.properties import Properties, PropertyFile
 from sagemaker.workflow.retry import RetryPolicy
 
 
@@ -688,23 +686,24 @@ class CompilationStep(Step):
     """Compilation step for workflow."""
 
     def __init__(
-            self,
-            name: str,
-            estimator: EstimatorBase,
-            model: Model,
-            inputs: CompilationInput = None,
-            job_arguments: List[str] = None,
-            depends_on: Union[List[str], List[Step]] = None,
-            retry_policies: List[RetryPolicy] = None,
-            display_name: str = None,
-            description: str = None,
-            cache_config: CacheConfig = None
+        self,
+        name: str,
+        estimator: EstimatorBase,
+        model: Model,
+        inputs: CompilationInput = None,
+        job_arguments: List[str] = None,
+        depends_on: Union[List[str], List[Step]] = None,
+        retry_policies: List[RetryPolicy] = None,
+        display_name: str = None,
+        description: str = None,
+        cache_config: CacheConfig = None,
     ):
-        """Construct a CompilationStep, given an given an `EstimatorBase` instance and
-        a `sagemaker.model.Model` instance.
+        """Construct a CompilationStep.
+
+        Given an `EstimatorBase` and a `sagemaker.model.Model` instance construct a CompilationStep.
 
-        In addition to the estimator and Model instances, the other arguments are those that are supplied to
-        the `compile_model` method of the `sagemaker.model.Model.compile_model`.
+        In addition to the estimator and Model instances, the other arguments are those that are
+        supplied to the `compile_model` method of the `sagemaker.model.Model.compile_model`.
 
         Args:
             name (str): The name of the compilation step.