From 5eaf4748c711fbcfe899743ae849422c7e8f7b38 Mon Sep 17 00:00:00 2001 From: Gui Heise Date: Wed, 19 Jan 2022 10:58:09 -0500 Subject: [PATCH] Fix helper names --- mev_inspect/classifiers/helpers.py | 4 ++-- mev_inspect/classifiers/specs/aave.py | 6 +++--- mev_inspect/classifiers/specs/compound.py | 8 ++++---- 3 files changed, 9 insertions(+), 9 deletions(-) diff --git a/mev_inspect/classifiers/helpers.py b/mev_inspect/classifiers/helpers.py index 2531b63..b09bee3 100644 --- a/mev_inspect/classifiers/helpers.py +++ b/mev_inspect/classifiers/helpers.py @@ -182,7 +182,7 @@ def _filter_transfers( return filtered_transfers -def _get_received_transfer( +def get_received_transfer( liquidator: str, child_transfers: List[Transfer] ) -> Optional[Transfer]: """Get transfer from AAVE to liquidator""" @@ -194,7 +194,7 @@ def _get_received_transfer( return None -def _get_debt_transfer( +def get_debt_transfer( liquidator: str, child_transfers: List[Transfer] ) -> Optional[Transfer]: """Get transfer from liquidator to AAVE""" diff --git a/mev_inspect/classifiers/specs/aave.py b/mev_inspect/classifiers/specs/aave.py index af6c7ff..9e7c836 100644 --- a/mev_inspect/classifiers/specs/aave.py +++ b/mev_inspect/classifiers/specs/aave.py @@ -1,6 +1,6 @@ from typing import List, Optional -from mev_inspect.classifiers.helpers import _get_debt_transfer, _get_received_transfer +from mev_inspect.classifiers.helpers import get_debt_transfer, get_received_transfer from mev_inspect.schemas.classifiers import ( ClassifiedTrace, ClassifierSpec, @@ -30,9 +30,9 @@ class AaveLiquidationClassifier(LiquidationClassifier): debt_purchase_amount = None received_amount = None - debt_transfer = _get_debt_transfer(liquidator, child_transfers) + debt_transfer = get_debt_transfer(liquidator, child_transfers) - received_transfer = _get_received_transfer(liquidator, child_transfers) + received_transfer = get_received_transfer(liquidator, child_transfers) if debt_transfer is not None and received_transfer is not None: diff --git a/mev_inspect/classifiers/specs/compound.py b/mev_inspect/classifiers/specs/compound.py index 66de624..7716869 100644 --- a/mev_inspect/classifiers/specs/compound.py +++ b/mev_inspect/classifiers/specs/compound.py @@ -2,8 +2,8 @@ from typing import List, Optional from mev_inspect.classifiers.helpers import ( CETH_TOKEN_ADDRESS, - _get_debt_transfer, - _get_received_transfer, + get_debt_transfer, + get_received_transfer, ) from mev_inspect.schemas.classifiers import ( Classification, @@ -41,9 +41,9 @@ class CompoundLiquidationClassifier(LiquidationClassifier): else liquidation_trace.inputs["repayAmount"] ) - debt_transfer = _get_debt_transfer(liquidator, child_transfers) + debt_transfer = get_debt_transfer(liquidator, child_transfers) - received_transfer = _get_received_transfer(liquidator, child_transfers) + received_transfer = get_received_transfer(liquidator, child_transfers) seize_trace = _get_seize_call(child_traces)