fix: inference handler issue (#105)

* fix: add missing parameter to warmup_fn

* trigger build

Author: Qingzi-Lan

README.md

@@ -7,7 +7,6 @@
 Serve machine learning models within a Docker container using Amazon
 SageMaker.
 
-
 ## :books: Background
 
 [Amazon SageMaker](https://aws.amazon.com/sagemaker/) is a fully managed service for data science and machine learning (ML) workflows.

src/sagemaker_inference/transformer.py

@@ -161,7 +161,7 @@ def validate_and_initialize(self, model_dir=environment.model_dir):  # type: ()
                 self._pre_model_fn(model_dir)
             self._model = self._model_fn(model_dir)
             if self._model_warmup_fn is not None:
-                self._model_warmup_fn(model_dir)
+                self._model_warmup_fn(model_dir, self._model)
             self._initialized = True
 
     def _validate_user_module_and_set_functions(self):