added shape to entity

2025-12-18 19:47:27 +01:00
parent c1c6c610ad
commit 6664ebdb9b
1 changed files with 1 additions and 0 deletions
@@ -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