fix: address review comments (iteration #2)

Author: aviruthen

sagemaker-mlops/tests/unit/workflow/test_steps.py

@@ -383,7 +383,12 @@ def test_step_validate_json_get_function_with_property_file():
 
     step_map = {"processing-step": processing_step}
 
-    Step._validate_json_get_function(step, json_get, step_map)
+        Step._validate_json_get_function(step, json_get, step_map)
+
+
+def test_step_find_dependencies_in_step_arguments_with_json_get():
+    from sagemaker.mlops.workflow.steps import Step, StepTypeEnum
+    from sagemaker.core.workflow.functions import JsonGet
 
 
 
@@ -475,6 +480,7 @@ def test_step_find_dependencies_in_step_arguments_with_delayed_return():
 
 
 
+
 def test_step_find_dependencies_in_step_arguments_with_string_reference():
     from sagemaker.mlops.workflow.steps import Step
     from sagemaker.core.helper.pipeline_variable import PipelineVariable