From 2926f0bd21d16a48eef0974e673c117339ba0c08 Mon Sep 17 00:00:00 2001 From: Aaron Berdy Date: Wed, 8 Dec 2021 10:37:54 -0800 Subject: [PATCH] fix: remove extraneous reference from local job container setup --- .../jobs/local/local_job_container_setup.py | 15 --------------- .../jobs/local/test_local_job_container_setup.py | 2 -- 2 files changed, 17 deletions(-) diff --git a/src/braket/jobs/local/local_job_container_setup.py b/src/braket/jobs/local/local_job_container_setup.py index 0f7eac021..0adc9042d 100644 --- a/src/braket/jobs/local/local_job_container_setup.py +++ b/src/braket/jobs/local/local_job_container_setup.py @@ -44,7 +44,6 @@ def setup_container( run_environment_variables.update( _get_env_script_mode_config(creation_kwargs["algorithmSpecification"]["scriptModeConfig"]) ) - run_environment_variables.update(_get_env_additional_lib()) run_environment_variables.update(_get_env_default_vars(aws_session, **creation_kwargs)) if _copy_hyperparameters(container, **creation_kwargs): run_environment_variables.update(_get_env_hyperparameters()) @@ -113,20 +112,6 @@ def _get_env_script_mode_config(script_mode_config: Dict[str, str]) -> Dict[str, return result -def _get_env_additional_lib() -> Dict[str, str]: - """For preview, we have some libraries that are not available publicly (yet). The container - will install these libraries if we set this env variable. - - Returns: - (Dict[str, str]): The set of key/value pairs that should be added as environment variables - to the running container. - """ - return { - "AMZN_BRAKET_IMAGE_SETUP_SCRIPT": "s3://amazon-braket-external-assets-preview-us-west-2/" - "HybridJobsAccess/scripts/setup-container.sh", - } - - def _get_env_default_vars(aws_session: AwsSession, **creation_kwargs) -> Dict[str, str]: """This function gets the remaining 'simple' env variables, that don't require any additional logic to determine what they are or when they should be added as env variables. diff --git a/test/unit_tests/braket/jobs/local/test_local_job_container_setup.py b/test/unit_tests/braket/jobs/local/test_local_job_container_setup.py index 93e114ef6..51199055f 100644 --- a/test/unit_tests/braket/jobs/local/test_local_job_container_setup.py +++ b/test/unit_tests/braket/jobs/local/test_local_job_container_setup.py @@ -81,8 +81,6 @@ def expected_envs(): return { "AMZN_BRAKET_CHECKPOINT_DIR": "/opt/omega/checkpoints", "AMZN_BRAKET_DEVICE_ARN": "test device ARN", - "AMZN_BRAKET_IMAGE_SETUP_SCRIPT": "s3://amazon-braket-external-assets-preview-us-west-2/" - "HybridJobsAccess/scripts/setup-container.sh", "AMZN_BRAKET_JOB_NAME": "Test-Job-Name", "AMZN_BRAKET_JOB_RESULTS_DIR": "/opt/braket/model", "AMZN_BRAKET_JOB_RESULTS_S3_PATH": "test_location/Test-Job-Name/output",