def _is_model_file()

in src/sagemaker_pytorch_serving_container/default_pytorch_inference_handler.py [0:0]

• 6 lines of code
• 2 McCabe index (conditional complexity)

    def _is_model_file(filename):
        is_model_file = False
        if os.path.isfile(filename):
            _, ext = os.path.splitext(filename)
            is_model_file = ext in [".pt", ".pth"]
        return is_model_file