From c8b80dddb04b92d6fe760e1da1600ae034a1a709 Mon Sep 17 00:00:00 2001 From: Scott Sanderson Date: Tue, 12 Jan 2016 17:37:28 -0500 Subject: [PATCH] BUG: Handle unicode adjustments path in py2. In Python 2, passing unicode to SQLiteAdjustmentReader would fail to coerce. --- zipline/data/us_equity_pricing.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/zipline/data/us_equity_pricing.py b/zipline/data/us_equity_pricing.py index 236dae5b..89e2c677 100644 --- a/zipline/data/us_equity_pricing.py +++ b/zipline/data/us_equity_pricing.py @@ -45,10 +45,11 @@ from pandas import ( ) from six import ( iteritems, - string_types, with_metaclass, ) +from zipline.utils.input_validation import coerce_string, preprocess + from ._equities import _compute_row_slices, _read_bcolz_data from ._adjustments import load_adjustments_from_sqlite @@ -910,9 +911,8 @@ class SQLiteAdjustmentReader(object): Connection from which to load data. """ + @preprocess(conn=coerce_string(sqlite3.connect)) def __init__(self, conn): - if isinstance(conn, str): - conn = sqlite3.connect(conn) self.conn = conn def load_adjustments(self, columns, dates, assets):