From 9025f54434d24608e3c91608c893e2f32f2e6861 Mon Sep 17 00:00:00 2001
From: Jens Ahrensfeld <jens@jayfield.org>
Date: Tue, 6 Jan 2026 21:49:35 +0100
Subject: [PATCH] simplified EntityParams.from_dict

---
 src/rbm/entity.py | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/src/rbm/entity.py b/src/rbm/entity.py
index 734f37e..1e78249 100644
--- a/src/rbm/entity.py
+++ b/src/rbm/entity.py
@@ -47,12 +47,8 @@ class EntityParams:
 	@classmethod
 	def from_dict(cls, params: dict):
 		obj = EntityParams()
-		if "doGaussianVisible" in params:
-			obj.do_gaussian_visible = params["doGaussianVisible"]
-		if "doGaussianHidden" in params:
-			obj.do_gaussian_hidden = params["doGaussianHidden"]
-		if "num_gibbs_samples" in params:
-			obj.num_gibbs_samples = params["num_gibbs_samples"]
+		for key, value in params.items():
+			setattr(obj, key, value)
 
 		return obj