From c93ae8193245d3cf2fb4c01babcb417524cb0564 Mon Sep 17 00:00:00 2001 From: Luke Van Seters Date: Thu, 15 Jul 2021 16:54:43 -0400 Subject: [PATCH] Use type not internal type --- mev_inspect/schemas/abi.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/mev_inspect/schemas/abi.py b/mev_inspect/schemas/abi.py index d337480..3c54837 100644 --- a/mev_inspect/schemas/abi.py +++ b/mev_inspect/schemas/abi.py @@ -6,8 +6,6 @@ from hexbytes import HexBytes from pydantic import BaseModel from web3 import Web3 -from .utils import CamelModel - class ABIDescriptionType(str, Enum): function = "function" @@ -25,8 +23,8 @@ NON_FUNCTION_DESCRIPTION_TYPES = Union[ ] -class ABIDescriptionInput(CamelModel): - internal_type: str +class ABIDescriptionInput(BaseModel): + type: str class ABIGenericDescription(BaseModel): @@ -43,8 +41,7 @@ class ABIFunctionDescription(BaseModel): return Web3.sha3(text=signature)[0:4] def get_signature(self) -> str: - joined_input_types = ",".join(input.internal_type for input in self.inputs) - + joined_input_types = ",".join(input.type for input in self.inputs) return f"{self.name}({joined_input_types})"