extend model member detection by list[Entity]

This commit is contained in:
2026-06-03 22:13:57 +02:00
parent 27460c503d
commit 5c3734a2a4
+4
View File
@@ -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):