diff --git a/fastlane_bot/modes/triangle_multi_complete.py b/fastlane_bot/modes/triangle_multi_complete.py index d053ae0fc..e5e8efa00 100644 --- a/fastlane_bot/modes/triangle_multi_complete.py +++ b/fastlane_bot/modes/triangle_multi_complete.py @@ -10,9 +10,9 @@ """ from typing import List, Any, Tuple, Union +from arb_optimizer import CurveContainer, MargPOptimizer + from fastlane_bot.modes.base_triangle import ArbitrageFinderTriangleBase -from fastlane_bot.tools.cpc import CPCContainer -from fastlane_bot.tools.optimizer import MargPOptimizer class ArbitrageFinderTriangleMultiComplete(ArbitrageFinderTriangleBase): @@ -34,7 +34,7 @@ def find_arbitrage(self, candidates: List[Any] = None, ops: Tuple = None, best_p for src_token, miniverse in combos: try: - CC_cc = CPCContainer(miniverse) + CC_cc = CurveContainer(miniverse) O = MargPOptimizer(CC_cc) pstart = self.build_pstart(CC_cc, CC_cc.tokens(), src_token) r = O.optimize(src_token, params=dict(pstart=pstart))