Fix helper names

This commit is contained in:
Gui Heise 2022-01-19 10:58:09 -05:00
parent e1d054b82e
commit 5eaf4748c7
3 changed files with 9 additions and 9 deletions

View File

@ -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"""

View File

@ -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:

View File

@ -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)