fix: fix lint and unused import + verify depends on step instance in integ test

Author: jerrypeng7773

src/sagemaker/workflow/steps.py

@@ -91,12 +91,7 @@ def to_request(self) -> RequestType:
         return request_dict
 
     def add_depends_on(self, step_names: Union[List[str], List["Step"]]):
-        """
-        Add step names to the current step depends on list
-
-        Args:
-            step_names (List[str] or List[Step]): A list of step name strings or step instances
-        """
+        """Add step names or stpe instances to the current step depends on list"""
 
         if not step_names:
             return

tests/integ/test_workflow.py

@@ -1083,7 +1083,7 @@ def test_two_processing_job_depends_on(
 
     step_pyspark_2 = ProcessingStep(
         name="pyspark-process-2",
-        depends_on=[step_pyspark_1.name],
+        depends_on=[step_pyspark_1],
         processor=pyspark_processor,
         inputs=spark_run_args.inputs,
         outputs=spark_run_args.outputs,

tests/integ/test_workflow_with_clarify.py

@@ -33,7 +33,8 @@
 from sagemaker.processing import ProcessingInput, ProcessingOutput
 from sagemaker.session import get_execution_role
 from sagemaker.workflow.conditions import ConditionLessThanOrEqualTo
-from sagemaker.workflow.condition_step import ConditionStep, JsonGet
+from sagemaker.workflow.condition_step import ConditionStep
+from sagemaker.workflow.functions import JsonGet
 from sagemaker.workflow.parameters import (
     ParameterInteger,
     ParameterString,