From 89ab65c413b65f7b38ff76bb1f2fa723acaae1ee Mon Sep 17 00:00:00 2001 From: Thomas Wiecki Date: Thu, 14 Mar 2013 08:56:16 -0400 Subject: [PATCH] REF: Import interface. Makes TradingAlgorithm available at the top-level. So that algorithms can do: ``` import zipline class MyAlgo(zipline.TradingAlgorithm) ``` OR ``` from zipline import TradingAlgorithm class MyAlgo(TradingAlgorithm) ``` --- zipline/__init__.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/zipline/__init__.py b/zipline/__init__.py index d0d7b7ac..a20d73f9 100644 --- a/zipline/__init__.py +++ b/zipline/__init__.py @@ -12,10 +12,13 @@ import finance import gens import utils +from algorithm import TradingAlgorithm + __all__ = [ 'ndict', 'data', 'finance', 'gens', - 'utils' + 'utils', + 'TradingAlgorithm' ]