pathways: do not create scripts for bootstrap and invoke_function modules (aws#826)

Co-authored-by: Rohan Gujarathi <[email protected]>

Author: 2 people

setup.py

@@ -107,8 +107,6 @@ def read_requirements(filename):
     entry_points={
         "console_scripts": [
             "sagemaker-upgrade-v2=sagemaker.cli.compatibility.v2.sagemaker_upgrade_v2:main",
-            "invoke-remote-function=sagemaker.remote_function.invoke_function:main",
-            "bootstrap-runtime-env=sagemaker.remote_function.bootstrap:main",
         ]
     },
 )

src/sagemaker/remote_function/invoke_function.py

@@ -33,10 +33,9 @@ def _parse_agrs():
     parser.add_argument("--region", type=str, required=True)
     parser.add_argument("--s3_base_uri", type=str, required=True)
     parser.add_argument("--s3_kms_key", type=str)
-    parser.add_argument("--dependencies", type=str)
-    parser.add_argument("--job_conda_env", type=str)
 
-    return parser.parse_args()
+    args, _ = parser.parse_known_args()
+    return args
 
 
 def _get_sagemaker_session(region):

src/sagemaker/remote_function/job_driver.sh

@@ -5,7 +5,7 @@
 set -eu
 
 printf "INFO: Bootstraping runtime environment\n"
-bootstrap-runtime-env "$@"
+python -m sagemaker.remote_function.bootstrap "$@"
 
 printf "INFO: Invoking remote function\n"
-invoke-remote-function "$@"
+python -m sagemaker.remote_function.invoke_function "$@"