From 93e22aa8b7a843ddcbc62b177f0f14d7f4b3a058 Mon Sep 17 00:00:00 2001 From: fawce Date: Wed, 20 Mar 2013 02:21:44 -0400 Subject: [PATCH] added more acceptable number types for event fields in batch transforms. --- zipline/transforms/utils.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/zipline/transforms/utils.py b/zipline/transforms/utils.py index 2c44dbcd..02d39bad 100644 --- a/zipline/transforms/utils.py +++ b/zipline/transforms/utils.py @@ -19,6 +19,7 @@ Generator versions of transforms. """ import types import logbook +import numpy from copy import deepcopy from datetime import datetime @@ -447,7 +448,13 @@ class BatchTransform(EventWindow): sid_keys = [] for sid in event.data.itervalues(): keys = set([name for name, value in sid.items() - if (isinstance(value, (int, float)))]) + if isinstance(value, + (int, + float, + numpy.integer, + numpy.float, + numpy.long)) + ]) sid_keys.append(keys) # with CUSTOM data events, there may be different fields