From 1e7158936144449f40e18d036f6081823fe0fd58 Mon Sep 17 00:00:00 2001
From: dmt <>
Date: Mon, 28 Oct 2019 21:25:15 +0100
Subject: [PATCH] Check user input for StackIteration param.

---
 cml/shared/parameter.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/cml/shared/parameter.py b/cml/shared/parameter.py
index 076b31a..1dfaab6 100644
--- a/cml/shared/parameter.py
+++ b/cml/shared/parameter.py
@@ -149,7 +149,13 @@ class StackIterations:
         return self.stack_iterations
 
     def __set__(self, instance, value):
-        self.stack_iterations = value
+        if isinstance(value, int):
+            self.stack_iterations = value
+        elif isinstance(value, str):
+            try:
+               self.stack_iterations = int(value)
+            except Exception as error:
+                print(error)
 
 
 class LearnblockMinimum:
-- 
GitLab