decorators: fix class/instance logic

This commit is contained in:
Robert Jordens 2015-04-03 14:55:20 -06:00 committed by Sebastien Bourdeauducq
parent 4091af69fd
commit e702fb7727
1 changed files with 4 additions and 3 deletions

View File

@ -1,6 +1,7 @@
import warnings import warnings
from migen.fhdl.structure import * from migen.fhdl.structure import *
from migen.fhdl.module import Module
from migen.fhdl.tools import insert_reset, rename_clock_domain from migen.fhdl.tools import insert_reset, rename_clock_domain
class ModuleTransformer: class ModuleTransformer:
@ -40,10 +41,10 @@ class ModuleTransformer:
return victim return victim
def __call__(self, victim): def __call__(self, victim):
try: if isinstance(victim, Module):
return self.wrap_class(victim)
except TypeError:
return self.wrap_instance(victim) return self.wrap_instance(victim)
else:
return self.wrap_class(victim)
@classmethod @classmethod
def adhoc(cls, i, *args, **kwargs): def adhoc(cls, i, *args, **kwargs):