def load(self) -> bool:
model_path = kserve.Storage.download(self.model_dir)
paths = [os.path.join(model_path, MODEL_BASENAME + model_extension)
for model_extension in MODEL_EXTENSIONS]
for path in paths:
if os.path.exists(path):
self._gateway = launch_gateway()
self._backend = Py4JBackend(self._gateway)
self.evaluator = make_evaluator(self._backend, path).verify()
self.input_fields = [inputField.getName() for inputField in self.evaluator.getInputFields()]
self.ready = True
break
return self.ready
def setUp(self):
self.gateway = launch_gateway()