diff --git a/zipline/pipeline/__init__.py b/zipline/pipeline/__init__.py index 89f0a33b..de6d142f 100644 --- a/zipline/pipeline/__init__.py +++ b/zipline/pipeline/__init__.py @@ -1,10 +1,10 @@ from __future__ import print_function from zipline.assets import AssetFinder -from .classifiers import Classifier +from .classifiers import Classifier, CustomClassifier from .engine import SimplePipelineEngine from .factors import Factor, CustomFactor -from .filters import Filter +from .filters import Filter, CustomFilter from .term import Term from .graph import TermGraph from .pipeline import Pipeline @@ -53,6 +53,8 @@ def engine_from_files(daily_bar_path, __all__ = ( 'Classifier', 'CustomFactor', + 'CustomFilter', + 'CustomClassifier', 'engine_from_files', 'Factor', 'Filter', diff --git a/zipline/pipeline/filters/__init__.py b/zipline/pipeline/filters/__init__.py index 2f18840c..fc88247a 100644 --- a/zipline/pipeline/filters/__init__.py +++ b/zipline/pipeline/filters/__init__.py @@ -1,7 +1,14 @@ -from .filter import Filter, NumExprFilter, NullFilter, PercentileFilter +from .filter import ( + CustomFilter, + Filter, + NumExprFilter, + NullFilter, + PercentileFilter, +) from .latest import Latest __all__ = [ + 'CustomFilter', 'Filter', 'Latest', 'NumExprFilter',