From bf4494702425cc6f3652016b598181f658207bd9 Mon Sep 17 00:00:00 2001
From: brian <schnitzelen@gmail.com>
Date: Sat, 19 Oct 2024 19:15:41 +0000
Subject: [PATCH] added utility script

---
 model/src/utils/__init__.py  |  1 +
 model/src/utils/get_class.py | 10 ++++++++++
 2 files changed, 11 insertions(+)
 create mode 100644 model/src/utils/get_class.py

diff --git a/model/src/utils/__init__.py b/model/src/utils/__init__.py
index 54c4ddf..56ac462 100644
--- a/model/src/utils/__init__.py
+++ b/model/src/utils/__init__.py
@@ -1,2 +1,3 @@
+from .get_class import get_class
 from .load_model import DEVICE, load_model
 from .vcda_dataset import VCDADataset
diff --git a/model/src/utils/get_class.py b/model/src/utils/get_class.py
new file mode 100644
index 0000000..3b245b0
--- /dev/null
+++ b/model/src/utils/get_class.py
@@ -0,0 +1,10 @@
+from importlib import import_module
+from types import ModuleType
+
+
+def get_class(path: str) -> ModuleType:
+    parts = path.split('.')
+    module_path = '.'.join(parts[:-1])
+    class_name = parts[-1]
+    module = import_module(module_path)
+    return getattr(module, class_name)