def _setup(self) -> None:
    self._model = load(self._tag, model_store=self.model_store)
    raw_predict_fn = getattr(self._model, self._predict_fn_name)
    self._predict_fn = functools.partial(raw_predict_fn, **self._partial_kwargs)
def _setup(self) -> None:
    self._model = load(self._tag, model_store=self.model_store)
    raw_predict_fn = getattr(self._model, self._predict_fn_name)
    self._predict_fn = functools.partial(raw_predict_fn, **self._partial_kwargs)