:fix: feature initialization

linkdd · Jul 20, 2016 · ba6d8c7 · ba6d8c7
1 parent 0a54e46
commit ba6d8c7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/link/model/__init__.py b/link/model/__init__.py
@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
 
-__version__ = '0.8'
+__version__ = '0.9'
 
 CONF_BASE_PATH = 'link/model'
diff --git a/link/model/feature.py b/link/model/feature.py
@@ -32,17 +32,17 @@ def create_model(self, schema):
     def __call__(self, schemaname):
         schema = self.resolve_schema(schemaname)
         model = self.create_model(schema)
-        return lambda **kwargs: model(schemaname, self.middleware, **kwargs)
+        return lambda **kwargs: model(schemaname, self.obj, **kwargs)
 
 
 class Model(object):
     _DATA_ID = 'id'
 
-    def __init__(self, schemaname, middleware, **kwargs):
+    def __init__(self, schemaname, obj, **kwargs):
         super(Model, self).__init__()
 
         self._schemaname = schemaname
-        self._middleware = middleware
+        self._obj = obj
 
         for key in kwargs:
             self[key] = kwargs[key]