diff --git a/zipline/algorithm.py b/zipline/algorithm.py index d38bdee6..72a9365f 100644 --- a/zipline/algorithm.py +++ b/zipline/algorithm.py @@ -35,7 +35,7 @@ from zipline.utils.factory import create_simulation_parameters from zipline.transforms.utils import StatefulTransform from zipline.finance.slippage import ( VolumeShareSlippage, - FixedSlippage, + SlippageModel, transact_partial ) from zipline.finance.commission import PerShare, PerTrade @@ -403,7 +403,7 @@ class TradingAlgorithm(object): self.blotter.transact = transact def set_slippage(self, slippage): - if not isinstance(slippage, (VolumeShareSlippage, FixedSlippage)): + if not isinstance(slippage, SlippageModel): raise UnsupportedSlippageModel() if self.initialized: raise OverrideSlippagePostInit()