fix: code typo (#152)

Author: yanbasic

src/emd/constants.py

@@ -8,7 +8,7 @@
 CODEPIPELINE_ROLE_NAME_TEMPLATE = f"{ENV_STACK_NAME}-CodePipelineRole-{{region}}"
 CLOUDFORMATION_ROLE_NAME_TEMPLATE = f"{ENV_STACK_NAME}-CloudFormationRole-{{region}}"
 
-STACK_COMPLATE_STATUS_LIST = ['CREATE_COMPLETE','UPDATE_COMPLETE']
+STACK_COMPLETE_STATUS_LIST = ['CREATE_COMPLETE','UPDATE_COMPLETE']
 EMD_STACK_NOT_EXISTS_STATUS = 'NOT_EXISTS'
 EMD_STACK_OK_STATUS = 'OK'
 EMD_STACK_NOT_WORKING_STATUS = 'NOT_WORKING'
@@ -18,7 +18,7 @@
 EMD_MODELS_LOCAL_DIR_TEMPLATE = "emd_models/{model_id}"
 EMD_MODELS_S3_KEY_TEMPLATE = EMD_MODELS_LOCAL_DIR_TEMPLATE
 
-EMD_DEFAULT_PROFILE_PARH = "~/.emd_default_profile"
+EMD_DEFAULT_PROFILE_PATH = "~/.emd_default_profile"
 
 EMD_DEFAULT_CONTAINER_PREFIX = "emd"
 

src/emd/models/model.py

@@ -388,7 +388,7 @@ def get_image_push_account_id(self):
         return current_account_id
 
     def get_image_uri(
-        sef,
+        self,
         account_id,
         region,
         image_name,

src/emd/utils/aws_service_utils.py

@@ -14,7 +14,7 @@
     EMD_STACK_OK_STATUS,
     ENV_STACK_NAME,
     MODEL_STACK_NAME_PREFIX,
-    STACK_COMPLATE_STATUS_LIST,
+    STACK_COMPLETE_STATUS_LIST,
 )
 from emd.models import Service, Model
 from emd.models.utils.constants import (

src/emd/utils/profile_manager.py

@@ -1,5 +1,5 @@
 
-from emd.constants import EMD_DEFAULT_PROFILE_PARH
+from emd.constants import EMD_DEFAULT_PROFILE_PATH
 import os
 from emd.utils.logger_utils import get_logger
 
@@ -10,7 +10,7 @@
 
 
 class ProfileManager:
-    def __init__(self,profile_path = EMD_DEFAULT_PROFILE_PARH):
+    def __init__(self,profile_path = EMD_DEFAULT_PROFILE_PATH):
         self.profile_path = os.path.expanduser(profile_path)
 
     def write_default_profile_name_to_local(self,aws_profile_name: str):
@@ -26,6 +26,7 @@ def load_profile_name_from_local(self):
 
 
     def set_default_aws_profile_from_local(self):
+        profile_name = None
         try:
             profile_name = self.load_profile_name_from_local()
         except Exception as e:

src/pipeline/backend/huggingface/embedding/transformers_embedding_backend.py

@@ -9,7 +9,6 @@
 import torch
 from emd.constants import EMD_MODELS_LOCAL_DIR_TEMPLATE
 from emd.utils.logger_utils import get_logger
-import torch
 from threading import Thread
 import json
 from transformers import AutoModel
@@ -47,7 +46,7 @@ def start(self):
             )
         model_abs_path = os.path.abspath(model_dir)
 
-        # TODO add model iint args from model's definition
+        # TODO add model init args from model's definition
         torch_dtype = self.pretrained_model_init_kwargs.get("torch_dtype")
         if torch_dtype is not None:
             self.pretrained_model_init_kwargs['torch_dtype'] = {

src/pipeline/backend/huggingface/llm/transformer_llm_backend.py

@@ -10,7 +10,6 @@
 from emd.constants import EMD_MODELS_LOCAL_DIR_TEMPLATE
 from emd.utils.logger_utils import get_logger
 from transformers import AutoModelForCausalLM, AutoTokenizer
-import torch
 from transformers import TextIteratorStreamer
 from threading import Thread
 import json
@@ -50,7 +49,7 @@ def start(self):
             )
         model_abs_path = os.path.abspath(model_dir)
 
-        # TODO add model iint args from model's definition
+        # TODO add model init args from model's definition
         self.model = AutoModelForCausalLM.from_pretrained(
                 model_abs_path,
                 torch_dtype="auto",

src/pipeline/backend/huggingface/rerank/transformers_rerank_backend.py

@@ -9,7 +9,6 @@
 import torch
 from emd.constants import EMD_MODELS_LOCAL_DIR_TEMPLATE
 from emd.utils.logger_utils import get_logger
-import torch
 from threading import Thread
 import json
 from transformers import AutoModelForSequenceClassification
@@ -47,7 +46,7 @@ def start(self):
             )
         model_abs_path = os.path.abspath(model_dir)
 
-        # TODO add model iint args from model's definition
+        # TODO add model init args from model's definition
         torch_dtype = self.pretrained_model_init_kwargs.get("torch_dtype")
         if torch_dtype is not None:
             self.pretrained_model_init_kwargs['torch_dtype'] = {

src/pipeline/framework/fast_api/fast_api.py

@@ -72,7 +72,7 @@ def health():
     return "200 OK"
 
 # As sagemaker endpoint requires...
-@app.post("/invocations")
+@app.post("/v1/invocations")
 @app.post("/v1/chat/completions")
 @app.post("/v1/embeddings")
 @app.post("/v1/rerank")

src/pipeline/utils/aws_service_utils.py

@@ -5,7 +5,6 @@
 from .logger_utils import get_logger
 import json
 import hashlib
-import boto3
 from emd.utils.aws_service_utils import get_current_region
 logger = get_logger(__name__)
 
@@ -39,7 +38,7 @@ def get_account_id():
     return account_id
 
 def create_s3_bucket(bucket_name,region):
-    s3 = boto3.client('s3')
+    s3 = boto3.client('s3', region_name=region)
     try:
         s3.head_bucket(Bucket=bucket_name)
     except:
@@ -58,8 +57,6 @@ def create_s3_bucket(bucket_name,region):
             )
         except (s3.exceptions.BucketAlreadyOwnedByYou,):
             logger.info(f"bucket: {bucket_name} exists")
-        except Exception as e:
-            raise
 
 
 def get_role_create_template(