Skip to content

Commit 3184d11

Browse files
authored
Fixed python command for embedding generation processor
1 parent 742c551 commit 3184d11

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

sagemaker_pipelines/embedding_generation/embedding_generation_pipeline.ipynb

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -495,6 +495,7 @@
495495
"source": [
496496
"bucket_name_param = ParameterString(name=\"BucketName\", default_value=str(bucket_name))\n",
497497
"env_name_param = ParameterString(name=\"EnvName\", default_value=str(env_name))\n",
498+
"env_lang_param = ParameterString(name=\"PyProcessEnv\", default_value=str('/root/.conda/envs/claymodel/bin/python3'))\n",
498499
"s2_processing_lvl_param = ParameterString(name=\"S2ProcessingLevel\", default_value=\"l2a\")\n",
499500
"aoi_name_param = ParameterString(name=\"AoiName\", default_value=str(aoi_name))\n",
500501
"s2_grid_id_param = ParameterString(name=\"GridId\", default_value=str(tile_id))\n",
@@ -652,7 +653,7 @@
652653
"role = get_execution_role()\n",
653654
"\n",
654655
"embedding_gen_processor = ScriptProcessor(\n",
655-
" command=['python3'],\n",
656+
" command=[env_lang_param.default_value],\n",
656657
" image_uri=geofm_emb_gen_image_uri_param,\n",
657658
" role=role,\n",
658659
" instance_count=emb_generation_instance_count_param,\n",

0 commit comments

Comments
 (0)