From 1042c5441614c0d7387ef1ea95a84a190bc69ac3 Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Thu, 15 Aug 2024 15:38:07 +0200
Subject: [PATCH] model can report its own name
---
model_server/base/models.py | 5 +++++
model_server/base/session.py | 3 +--
2 files changed, 6 insertions(+), 2 deletions(-)
diff --git a/model_server/base/models.py b/model_server/base/models.py
index c24ed92d..ffaa346f 100644
--- a/model_server/base/models.py
+++ b/model_server/base/models.py
@@ -48,6 +48,11 @@ class Model(ABC):
def reload(self):
self.load()
+ @property
+ def name(self):
+ return f'{self.__class__}'
+
+
class ImageToImageModel(Model):
"""
diff --git a/model_server/base/session.py b/model_server/base/session.py
index 0e624ab1..8358a957 100644
--- a/model_server/base/session.py
+++ b/model_server/base/session.py
@@ -240,8 +240,7 @@ class _Session(object):
if key is None:
def mid(i):
- # TODO: give model the option to report its own name
- return f'{ModelClass.__name__}_{i:02d}'
+ return f'{mi.name}_{i:02d}'
while mid(ii) in self.models.keys():
ii += 1
--
GitLab