Changed SKLearnProcessor framework version back to 1.0-1

aws · Sep 2, 2022 · 0a899ea · 0a899ea
1 parent 8c07d0c
commit 0a899ea
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/sagemaker-pipelines/tabular/local-mode/sagemaker-pipelines-local-mode.ipynb b/sagemaker-pipelines/tabular/local-mode/sagemaker-pipelines-local-mode.ipynb
@@ -339,7 +339,7 @@
    "source": [
     "from sagemaker.sklearn.processing import SKLearnProcessor\n",
     "\n",
-    "framework_version = \"0.23-1\"\n",
+    "framework_version = \"1.0-1\"\n",
     "\n",
     "sklearn_processor = SKLearnProcessor(\n",
     "    framework_version=framework_version,\n",
@@ -1126,9 +1126,7 @@
    "source": [
     "# Get output from model evaluation step (processing job)\n",
     "\n",
-    "processing_job_name = steps[\"PipelineExecutionSteps\"][2][\"Metadata\"][\"ProcessingJob\"][\n",
-    "    \"Arn\"\n",
-    "]\n",
+    "processing_job_name = steps[\"PipelineExecutionSteps\"][2][\"Metadata\"][\"ProcessingJob\"][\"Arn\"]\n",
     "outputs = local_pipeline_session.sagemaker_client.describe_processing_job(\n",
     "    ProcessingJobName=processing_job_name\n",
     ")[\"ProcessingOutputConfig\"][\"Outputs\"]\n",
@@ -1158,9 +1156,7 @@
    "source": [
     "# Get output from the TransformStep\n",
     "\n",
-    "transform_job_name = steps[\"PipelineExecutionSteps\"][4][\"Metadata\"][\"TransformJob\"][\n",
-    "    \"Arn\"\n",
-    "]\n",
+    "transform_job_name = steps[\"PipelineExecutionSteps\"][4][\"Metadata\"][\"TransformJob\"][\"Arn\"]\n",
     "outputs = local_pipeline_session.sagemaker_client.describe_transform_job(\n",
     "    TransformJobName=transform_job_name\n",
     ")\n",
@@ -1195,7 +1191,7 @@
    "source": [
     "# Recreate the SKLearnProcessor with non-local session\n",
     "\n",
-    "framework_version = \"0.23-1\"\n",
+    "framework_version = \"1.0-1\"\n",
     "\n",
     "sklearn_processor = SKLearnProcessor(\n",
     "    framework_version=framework_version,\n",