From e4f47c0e42e6151a39d8068357ae672d4e5b31d5 Mon Sep 17 00:00:00 2001 From: Patrick Daly Date: Fri, 13 Aug 2021 14:14:14 -0700 Subject: [PATCH] correcting an import path --- tests/test_arbitrage_integration.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_arbitrage_integration.py b/tests/test_arbitrage_integration.py index ba2e760..210428f 100644 --- a/tests/test_arbitrage_integration.py +++ b/tests/test_arbitrage_integration.py @@ -1,5 +1,5 @@ -from mev_inspect.classifier_specs import CLASSIFIER_SPECS -from mev_inspect.trace_classifier import TraceClassifier +from mev_inspect.classifiers.specs import ALL_CLASSIFIER_SPECS +from mev_inspect.classifiers.trace import TraceClassifier from mev_inspect.arbitrage import get_arbitrages from mev_inspect.swaps import get_swaps @@ -9,7 +9,7 @@ from .utils import load_test_block def test_arbitrage_real_block(): block = load_test_block(12914994) - trace_clasifier = TraceClassifier(CLASSIFIER_SPECS) + trace_clasifier = TraceClassifier(ALL_CLASSIFIER_SPECS) classified_traces = trace_clasifier.classify(block.traces) swaps = get_swaps(classified_traces)