Use classmethod syntax

phanrahan · Jun 10, 2019 · b0d19fe · b0d19fe
1 parent e019932
commit b0d19fe
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/magma/array.py b/magma/array.py
@@ -257,8 +257,9 @@ def const(self):
     def flatten(self):
         return sum([t.flatten() for t in self.ts], [])
 
-    def concat(self, *args):
-        return concat(self, *args)
+    @classmethod
+    def concat(cls, *args):
+        return concat(*args)
 
 
 # def Array(N, T):

diff --git a/tests/test_type/test_bits.py b/tests/test_type/test_bits.py
@@ -110,7 +110,7 @@ class TestCircuit(m.Circuit):
         @classmethod
         def definition(io):
             a = io.I
-            b = a[0:-1].concat(m.bits(0, 1))
+            b = m.BFloat.concat(a[0:-1], m.bits(0, 1))
             io.O <= b
     print(repr(TestCircuit))
     assert repr(TestCircuit) == """\