fix ei fixture

Author: laurenyu

tests/integ/test_pytorch.py

@@ -160,18 +160,15 @@ def test_deploy_packed_model_with_entry_point_name(
     test_region() not in EI_SUPPORTED_REGIONS, reason="EI isn't supported in that specific region."
 )
 def test_deploy_model_with_accelerator(
-    sagemaker_session,
-    cpu_instance_type,
-    pytorch_eia_latest_ei_version,
-    pytorch_eia_latest_py_version,
+    sagemaker_session, cpu_instance_type, pytorch_eia_latest_version, pytorch_eia_latest_py_version,
 ):
     endpoint_name = "test-pytorch-deploy-eia-{}".format(sagemaker_timestamp())
     model_data = sagemaker_session.upload_data(path=EIA_MODEL)
     pytorch = PyTorchModel(
         model_data,
         "SageMakerRole",
         entry_point=EIA_SCRIPT,
-        framework_version=pytorch_eia_latest_ei_version,
+        framework_version=pytorch_eia_latest_version,
         py_version=pytorch_eia_latest_py_version,
         sagemaker_session=sagemaker_session,
     )