diff --git a/test/integ_tests/job_test_script.py b/test/integ_tests/job_test_script.py index 1fd7a364f..7071ffd62 100644 --- a/test/integ_tests/job_test_script.py +++ b/test/integ_tests/job_test_script.py @@ -11,19 +11,19 @@ # ANY KIND, either express or implied. See the License for the specific # language governing permissions and limitations under the License. +import json +import os + from braket.aws import AwsDevice from braket.circuits import Circuit -from braket.jobs import ( - get_hyperparameters, - get_job_device_arn, - save_job_checkpoint, - save_job_result, -) +from braket.jobs import save_job_checkpoint, save_job_result from braket.jobs_data import PersistedJobDataFormat def start_here(): - hyperparameters = get_hyperparameters() + hp_file = os.environ["AMZN_BRAKET_HP_FILE"] + with open(hp_file, "r") as f: + hyperparameters = json.load(f) if hyperparameters["test_case"] == "completed": completed_job_script() @@ -40,7 +40,7 @@ def completed_job_script(): print("Test job started!!!!!") # Use the device declared in the Orchestration Script - device = AwsDevice(get_job_device_arn()) + device = AwsDevice(os.environ["AMZN_BRAKET_DEVICE_ARN"]) bell = Circuit().h(0).cnot(0, 1) for count in range(5):