udpate named

Author: jeniyat

src/sagemaker/model.py

@@ -41,7 +41,7 @@
 LOGGER = logging.getLogger("sagemaker")
 
 NEO_ALLOWED_FRAMEWORKS = set(
-    ["mxnet", "tensorflow", "keras", "pytorch", "onnx", "xgboost", "tflite", "huggingface"]
+    ["mxnet", "tensorflow", "keras", "pytorch", "onnx", "xgboost", "tflite"]
 )
 
 

tests/conftest.py

@@ -270,17 +270,17 @@ def huggingface_tensorflow_latest_training_py_version():
 
 
 @pytest.fixture(scope="module")
-def huggingface_neo_latest_inference_pytorch_version():
+def huggingface_neuron_latest_inference_pytorch_version():
     return "1.9"
 
 
 @pytest.fixture(scope="module")
-def huggingface_neo_latest_inference_transformer_version():
+def huggingface_neuron_latest_inference_transformer_version():
     return "4.12"
 
 
 @pytest.fixture(scope="module")
-def huggingface_neo_latest_inference_py_version():
+def huggingface_neuron_latest_inference_py_version():
     return "py37"
 
 

tests/unit/sagemaker/huggingface/test_estimator.py

@@ -252,21 +252,21 @@ def test_huggingface(
     assert actual_train_args == expected_train_args
 
 
-def test_huggingface_neo(
+def test_huggingface_neuron(
     sagemaker_session,
-    huggingface_neo_latest_inference_pytorch_version,
-    huggingface_neo_latest_inference_transformer_version,
-    huggingface_neo_latest_inference_py_version,
+    huggingface_neuron_latest_inference_pytorch_version,
+    huggingface_neuron_latest_inference_transformer_version,
+    huggingface_neuron_latest_inference_py_version,
 ):
 
     inputs = "s3://mybucket/train"
     huggingface_model = HuggingFaceModel(
         model_data=inputs,
-        transformers_version=huggingface_neo_latest_inference_transformer_version,
+        transformers_version=huggingface_neuron_latest_inference_transformer_version,
         role=ROLE,
         sagemaker_session=sagemaker_session,
-        pytorch_version=huggingface_neo_latest_inference_pytorch_version,
-        py_version=huggingface_neo_latest_inference_py_version,
+        pytorch_version=huggingface_neuron_latest_inference_pytorch_version,
+        py_version=huggingface_neuron_latest_inference_py_version,
     )
     container = huggingface_model.prepare_container_def("ml.inf.xlarge")
     assert container["Image"]