test21

Author: Qian Chen

buildspec.yml

@@ -29,10 +29,10 @@ phases:
       - PY_COLORS=0
 
       # run linters
-      # - tox -e flake8,twine
+      - tox -e flake8,twine
 
       # run unit tests
-      # - tox -e py36,py37,py38 test/unit
+      - tox -e py36,py37,py38 test/unit
 
       # define tags
       - GENERIC_CPU_TAG="$FRAMEWORK_VERSION-mxnet-cpu-$BUILD_ID"
@@ -41,8 +41,8 @@ phases:
       - DLC_EIA_TAG="$EIA_FRAMEWORK_VERSION-dlc-eia-$BUILD_ID"
 
       # run local CPU integration tests (build and push the image to ECR repo)
-      # - test_cmd="IGNORE_COVERAGE=- tox -e py36 -- test/integration/local --build-image --push-image --dockerfile-type mxnet.cpu --region $AWS_DEFAULT_REGION --docker-base-name $ECR_REPO --aws-id $ACCOUNT --framework-version $FRAMEWORK_VERSION --processor cpu --tag $GENERIC_CPU_TAG"
-      # - execute-command-if-has-matching-changes "$test_cmd" "test/" "src/*.py" "setup.py" "setup.cfg" "buildspec.yml" "artifacts/*"
+      - test_cmd="IGNORE_COVERAGE=- tox -e py36 -- test/integration/local --build-image --push-image --dockerfile-type mxnet.cpu --region $AWS_DEFAULT_REGION --docker-base-name $ECR_REPO --aws-id $ACCOUNT --framework-version $FRAMEWORK_VERSION --processor cpu --tag $GENERIC_CPU_TAG"
+      - execute-command-if-has-matching-changes "$test_cmd" "test/" "src/*.py" "setup.py" "setup.cfg" "buildspec.yml" "artifacts/*"
       - test_cmd="IGNORE_COVERAGE=- tox -e py36 -- test/integration/local --build-image --push-image --dockerfile-type dlc.cpu --region $AWS_DEFAULT_REGION --docker-base-name $ECR_REPO --aws-id $ACCOUNT --framework-version $FRAMEWORK_VERSION --processor cpu --tag $DLC_CPU_TAG"
       - execute-command-if-has-matching-changes "$test_cmd" "test/" "src/*.py" "setup.py" "setup.cfg" "buildspec.yml" "artifacts/*"
 

src/sagemaker_mxnet_serving_container/handler_service.py

@@ -47,22 +47,10 @@ def __init__(self):
 
     @staticmethod
     def _user_module_transformer(model_dir=environment.model_dir):
-        log.info('schenqian1')
-        log.info(model_dir)
         module_name = environment.Environment().module_name
-        log.info(module_name)
         inference_script = model_dir + '/code' + '/{}.py'.format(module_name)
-        log.info(inference_script)
         if os.path.exists(inference_script):
-            log.info('schenqian2')
-            try:
-                spec = importlib.util.spec_from_file_location(module_name, inference_script)
-            except Exception as e:
-                log.info('schenqian5')
-                log.info("{}".format(e))
-            if spec is None:
-                log.info('schenqian4')
-            log.info('schenqian3')
+            spec = importlib.util.spec_from_file_location(module_name, inference_script)
             user_module = importlib.util.module_from_spec(spec)
             log.info(user_module)
         else: