From 5747378d9a25fac1ac163db34d64264511fe5090 Mon Sep 17 00:00:00 2001 From: Martin Sivak Date: Wed, 28 Nov 2012 16:35:48 +0100 Subject: Add wraps call around class decorator in inject --- di/core.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/di/core.py b/di/core.py index db3202e..f41d601 100644 --- a/di/core.py +++ b/di/core.py @@ -133,8 +133,8 @@ def inject(*args, **kwargs): return obj if not isinstance(obj, DiRegistry): - obj = DiRegistry(obj) - + obj = wraps(obj)(DiRegistry(obj)) + obj.register(*args, **kwargs) return obj -- cgit