Fix helper names
This commit is contained in:
parent
e1d054b82e
commit
5eaf4748c7
@ -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"""
|
||||
|
@ -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:
|
||||
|
||||
|
@ -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)
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user