Merge pull request #997 from rizar/input_dim_for_lookup_table

Adds input_dim setter to LookupTable
mila-iqia · Feb 22, 2016 · aa07fd8 · aa07fd8
2 parents c25a5f0 + c67e11e
commit aa07fd8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/blocks/bricks/lookup.py b/blocks/bricks/lookup.py
@@ -78,6 +78,11 @@ def get_dim(self, name):
     def input_dim(self):
         return 0
 
+    @input_dim.setter
+    def input_dim(self, dim):
+        if dim != 0:
+            raise ValueError("LookupTable input must be integer")
+
     @property
     def output_dim(self):
         return self.dim

diff --git a/tests/bricks/test_lookup.py b/tests/bricks/test_lookup.py
@@ -35,4 +35,5 @@ def test_lookup_table():
 
     def assign_input_dim():
         lt.input_dim = 11
-    assert_raises(AttributeError, assign_input_dim)
+    assert_raises(ValueError, assign_input_dim)
+    lt.input_dim = 0