From 3e3cbf375eccdce63c6e12a6d5af409dd79d2b9b Mon Sep 17 00:00:00 2001
From: Thilo Huellmann <thilo@colabel.com>
Date: Tue, 18 Jan 2022 00:38:20 +0100
Subject: [PATCH] Hint at correct env var name in error message

HF_TASK instead of TASK
---
 src/sagemaker_huggingface_inference_toolkit/handler_service.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/sagemaker_huggingface_inference_toolkit/handler_service.py b/src/sagemaker_huggingface_inference_toolkit/handler_service.py
index a0e4d67..b7ac0b8 100644
--- a/src/sagemaker_huggingface_inference_toolkit/handler_service.py
+++ b/src/sagemaker_huggingface_inference_toolkit/handler_service.py
@@ -107,7 +107,7 @@ def load(self, model_dir):
             hf_pipeline = get_pipeline(task=task, model_dir=model_dir, device=self.device)
         else:
             raise ValueError(
-                f"You need to define one of the following {list(SUPPORTED_TASKS.keys())} as env 'TASK'.", 403
+                f"You need to define one of the following {list(SUPPORTED_TASKS.keys())} as env 'HF_TASK'.", 403
             )
         return hf_pipeline