Fix ff tests

phanrahan · Nov 9, 2018 · 186fda9 · 186fda9
1 parent 417d741
commit 186fda9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/magma/circuit.py b/magma/circuit.py
@@ -28,7 +28,6 @@
 __all__ += ['getCurrentDefinition']
 __all__ += ['magma_clear_circuit_cache']
 
-__all__ += ['isprimitive']
 __all__ += ['CopyInstance']
 __all__ += ['circuit_type_method']
 __all__ += ['circuit_generator']
@@ -390,9 +389,6 @@ def popDefinition():
     else:
         currentDefinition = None
 
-def isprimitive(circuit):
-    return getattr(circuit, "primitive", False)
-
 # a map from circuitDefinition names to circuit definition objects
 definitionCache = {}
 

diff --git a/magma/transforms.py b/magma/transforms.py
@@ -1,5 +1,7 @@
 from collections import namedtuple, OrderedDict
+from .circuit import DefineCircuit, EndCircuit, CopyInstance
 from .is_definition import isdefinition
+from .is_primitive import isprimitive
 from .bit import *
 from .clock import ClockType, EnableType, ResetType, AsyncResetType, wiredefaultclock
 from .array import *