wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-07-13 17:42:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix syntax in transform

Browse Source
This commit is contained in:
scottsanderson
2012-08-01 18:56:29 -04:00
parent 4768bea646
commit 9e1a5c11cb
1 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
zipline/gens/transform.py
+14 -4
View File
@@ -79,11 +79,21 @@ def stateful_transform(stream_in, tnfm_class, *args, **kwargs):
out_message.tnfm_value = tnfm_value
yield out_message
# Special logic for TransactionSimulator. This is ugly but I
# want to get to testing faster. Should be refactored later
# to something that doesn't make Scott cry.
elif tnfm_class.__name__ == 'TransactionSimulator'
# Special logic for TransactionSimulator and
# PerformanceTracker. This is ugly but I want to get to
# testing faster. Should be refactored later to something
# that doesn't make Scott cry.
elif tnfm_class.__name__ == 'TransactionSimulator':
out_message = message_copy
out_message.txn = tnfm_value
yield out_message
elif tnfm_class.__name__ == 'PerformanceTracker':
out_message = message_copy
del out_message['txn']
out_message.portfolio = tnfm_value
yield out_message
# Otherwise send tnfm_id, tnfm_value, and the message date.
else:
out_message = ndict()