From 1f40684566211a93b3c362acf1b8db87f3575880 Mon Sep 17 00:00:00 2001 From: scottsanderson Date: Sat, 28 Jul 2012 00:17:28 -0400 Subject: [PATCH] patch for filter logic --- zipline/gens/tradegens.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/zipline/gens/tradegens.py b/zipline/gens/tradegens.py index 2b2598e8..6404c403 100644 --- a/zipline/gens/tradegens.py +++ b/zipline/gens/tradegens.py @@ -37,7 +37,7 @@ def SpecificEquityTrades(count = 500, sids = [1, 2], event_list = None, filter = unfiltered = (create_trade(*args) for args in arg_gen) if filter: - filtered = ifilter(lambda event: event.sid in filter) + filtered = ifilter(lambda event: event.sid in filter, unfiltered) else: filtered = unfiltered @@ -46,4 +46,4 @@ def SpecificEquityTrades(count = 500, sids = [1, 2], event_list = None, filter = if __name__ == "__main__": import nose.tools; nose.tools.set_trace() - trades = SpecificEquityTrades() + trades = SpecificEquityTrades(filter = [1])