diff --git a/tests/pipeline/test_engine.py b/tests/pipeline/test_engine.py index 6ab43c39..01892a4c 100644 --- a/tests/pipeline/test_engine.py +++ b/tests/pipeline/test_engine.py @@ -39,7 +39,7 @@ from zipline.pipeline.loaders.equity_pricing_loader import ( USEquityPricingLoader, ) from zipline.pipeline.engine import SimplePipelineEngine -from zipline.pipeline.factor import CustomFactor +from zipline.pipeline import CustomFactor from zipline.pipeline.factor.technical import ( MaxDrawdown, SimpleMovingAverage, diff --git a/zipline/pipeline/__init__.py b/zipline/pipeline/__init__.py index c9eb0fa5..d2449630 100644 --- a/zipline/pipeline/__init__.py +++ b/zipline/pipeline/__init__.py @@ -1,5 +1,5 @@ from .classifier import Classifier -from .factor import Factor +from .factor import Factor, CustomFactor from .filter import Filter from .term import Term from .graph import TermGraph