Fix for Python 2.7 syntax, simplified native layer code.

aigamedev · Apr 2, 2016 · 6f84794 · 6f84794
1 parent 5df1253
commit 6f84794
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/sknn/nn.py b/sknn/nn.py
@@ -29,13 +29,9 @@ class ansi:
 
 class Native(object):
 
-    def __init__(self, constructor, *args, name=None, units=None, **keywords):
-        self.name = name
-        self.units = units
-
-        self.frozen = None
-        self.weight_decay = None
-        self.normalize = None
+    def __init__(self, constructor, *args, **keywords):
+        for attr in ['name', 'units', 'frozen', 'weight_decay', 'normalize']:
+            setattr(self, attr, keywords.pop(attr, None))
 
         self.type = constructor
         self.args = args