fixing code: permissions for non-root users, integration tests

Author: brunopistone

src/sagemaker/remote_function/client.py

@@ -92,7 +92,6 @@ def remote(
     max_wait_time_in_seconds=None,
     use_torchrun=False,
     nproc_per_node=1,
-
 ):
     """Decorator for running the annotated function as a SageMaker training job.
 

src/sagemaker/remote_function/runtime_environment/bootstrap_runtime_environment.py

@@ -43,7 +43,7 @@
 REMOTE_FUNCTION_WORKSPACE = "sm_rf_user_ws"
 BASE_CHANNEL_PATH = "/opt/ml/input/data"
 FAILURE_REASON_PATH = "/opt/ml/output/failure"
-JOB_OUTPUT_DIRS = ["/opt/ml/output", "/opt/ml/model", "/tmp"]
+JOB_OUTPUT_DIRS = ["/opt/ml/input", "/opt/ml/output", "/opt/ml/model", "/tmp"]
 PRE_EXECUTION_SCRIPT_NAME = "pre_exec.sh"
 JOB_REMOTE_FUNCTION_WORKSPACE = "sagemaker_remote_function_workspace"
 SCRIPT_AND_DEPENDENCIES_CHANNEL_NAME = "pre_exec_script_and_dependencies"