From 5c3734a2a4e2969e8e2c16e73ff6cad2ce14af4f Mon Sep 17 00:00:00 2001
From: Jens Ahrensfeld <jens@jayfield.org>
Date: Wed, 3 Jun 2026 22:13:57 +0200
Subject: [PATCH] extend model member detection by list[Entity]

---
 src/model/model.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/model/model.py b/src/model/model.py
index 6e43f6d..7a56bef 100644
--- a/src/model/model.py
+++ b/src/model/model.py
@@ -1,6 +1,8 @@
 import os
 from abc import ABC, abstractmethod
 
+from IPython.utils.wildcard import is_type
+
 from rbm.entity import Entity
 from rbm.train import train
 from rbm.matrix import Mat, np
@@ -18,6 +20,8 @@ class Model(ABC):
 		for name, value in self.__dict__.items():
 			if isinstance(value, obj_type):
 				obj_list.append(value)
+			elif isinstance(value, type(obj_list)):
+				obj_list += value
 		return obj_list
 
 	def train(self, batch: Mat, status: Status = None):