Remove unsed processor and fetch JsonGet from functions package

Author: Niklas Palm

04_training_pipeline/pipeline.ipynb

@@ -471,12 +471,9 @@
     "from sagemaker.processing import (\n",
     "    ProcessingInput,\n",
     "    ProcessingOutput,\n",
-    "    FrameworkProcessor,\n",
     "    ScriptProcessor,\n",
     ")\n",
     "\n",
-    "\n",
-    "\n",
     "eval_steps = dict()\n",
     "eval_reports = dict()\n",
     "\n",
@@ -599,10 +596,8 @@
    "outputs": [],
    "source": [
     "from sagemaker.workflow.conditions import ConditionGreaterThanOrEqualTo\n",
-    "from sagemaker.workflow.condition_step import (\n",
-    "    ConditionStep,\n",
-    "    JsonGet,\n",
-    ")\n",
+    "from sagemaker.workflow.condition_step import ConditionStep\n",
+    "from sagemaker.workflow.functions import JsonGet\n",
     "\n",
     "condition_steps = dict()\n",
     "\n",
@@ -612,7 +607,7 @@
     "    # Models with a test accuracy lower than the condition will not be registered with the model registry.\n",
     "    cond_gte = ConditionGreaterThanOrEqualTo(\n",
     "        left=JsonGet(\n",
-    "            step=eval_steps[t],\n",
+    "            step_name=eval_steps[t].name,\n",
     "            property_file=eval_reports[t],\n",
     "            json_path=\"multiclass_classification_metrics.accuracy.value\",\n",
     "        ),\n",