SetConst new dispatch

hanjinliu · Sep 5, 2021 · 248a674 · 248a674
1 parent 37e65a0
commit 248a674
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/impy/_const.py b/impy/_const.py
@@ -88,7 +88,13 @@ def __repr__(self):
 
 class SetConst:
     n_ongoing = 0
-    def __init__(self, name, value):
+    def __init__(self, name=None, value=None, **kwargs):
+        if name is None and value is None and len(kwargs) == 1:
+            name, value = list(kwargs.items())[0]
+        elif name in Const.keys() and value is not None:
+            pass
+        else:
+            raise TypeError("Invalid input for SetConst")
         self.name = name
         self.value = value