fhdl/decorators: make the transform logic more idiomatic

[litex.git] / migen / fhdl / simplify.py

diff --git a/migen/fhdl/simplify.py b/migen/fhdl/simplify.py
index 531146553e0e75346c4810a42725fb9bcce80ada..1bc2eeca0ebf8754e4d65c9bfc05d0a96560cc01 100644 (file)
--- a/migen/fhdl/simplify.py
+++ b/migen/fhdl/simplify.py
@@ -1,10 +1,10 @@
 from migen.fhdl.std import *
 from migen.fhdl.specials import _MemoryPort
-from migen.fhdl.decorators import ModuleDecorator
+from migen.fhdl.decorators import ModuleTransformer
 from migen.util.misc import gcd_multiple
 
-class FullMemoryWE(ModuleDecorator):
-       def transform_fragment(self, f):
+class FullMemoryWE(ModuleTransformer):
+       def transform_fragment(self, i, f):
                newspecials = set()
 
                for orig in f.specials: