Datawriter publishes feature version to the event log as well

JohnVinyard · Oct 13, 2017 · c7153f9 · c7153f9
1 parent 8c462df
commit c7153f9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/featureflow/__init__.py b/featureflow/__init__.py
@@ -1,4 +1,4 @@
-__version__ = '2.2.4'
+__version__ = '2.3.4'
 
 from model import BaseModel, ModelExistsError
 

diff --git a/featureflow/datawriter.py b/featureflow/datawriter.py
@@ -45,7 +45,11 @@ def __exit__(self, t, value, traceback):
         self._stream.close()
         if self.event_log is not None:
             self.event_log.append(
-                json.dumps({'_id': self._id, 'name': self.feature_name}))
+                json.dumps({
+                    '_id': self._id,
+                    'name': self.feature_name,
+                    'version': self.feature_version
+                }))
 
     def _process(self, data):
         yield self._stream.write(data)