From 6664ebdb9b62fa615ec8c8d05d570405dca75232 Mon Sep 17 00:00:00 2001
From: Jens Ahrensfeld <jens@jayfield.org>
Date: Thu, 18 Dec 2025 19:47:27 +0100
Subject: [PATCH] added shape to entity

---
 src/rbm/entity.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/rbm/entity.py b/src/rbm/entity.py
index a283c40..62e4473 100644
--- a/src/rbm/entity.py
+++ b/src/rbm/entity.py
@@ -6,6 +6,7 @@ from status import Status
 
 class Entity:
 	def __init__(self, shape: tuple[int, int], params: RbmParams):
+		self.shape = shape
 		self.state = RbmState.from_layer_params(shape)
 		self.params = params