diff --git a/mev_inspect/transfers.py b/mev_inspect/transfers.py index 865a54e..309d47c 100644 --- a/mev_inspect/transfers.py +++ b/mev_inspect/transfers.py @@ -33,15 +33,15 @@ def get_eth_transfers(traces: List[ClassifiedTrace]) -> List[Transfer]: def get_transfer(trace: ClassifiedTrace) -> Optional[Transfer]: if trace.value is not None and trace.value > 0: - return _build_eth_transfer(trace) + return _get_eth_transfer(trace) if isinstance(trace, DecodedCallTrace): - return _build_erc20_transfer(trace) + return _get_erc20_transfer(trace) return None -def _build_eth_transfer(trace: ClassifiedTrace) -> Transfer: +def _get_eth_transfer(trace: ClassifiedTrace) -> Transfer: return Transfer( block_number=trace.block_number, transaction_hash=trace.transaction_hash, @@ -53,7 +53,7 @@ def _build_eth_transfer(trace: ClassifiedTrace) -> Transfer: ) -def _build_erc20_transfer(trace: DecodedCallTrace) -> Optional[Transfer]: +def _get_erc20_transfer(trace: DecodedCallTrace) -> Optional[Transfer]: classifier = get_classifier(trace) if classifier is not None and issubclass(classifier, TransferClassifier): return classifier.get_transfer(trace)