diff --git a/model_server/session.py b/model_server/session.py index 58e802c2247a5914c2e3dd4240e7222a4f56e63f..73b000cf69d98ec070882770f40aedc3ac79b1e1 100644 --- a/model_server/session.py +++ b/model_server/session.py @@ -116,7 +116,7 @@ class Session(object): 'params': params } self.log_event(f'Loaded model {key}') - return key + return {key: self.models[key]} def describe_loaded_models(self) -> dict: return {