Move classifier spec to its own schemas file

mev_inspect/classifiers/specs/aave.py

@@ -1,6 +1,6 @@
+from mev_inspect.schemas.classifier_specs import ClassifierSpec
 from mev_inspect.schemas.classified_traces import (
     Classification,
-    ClassifierSpec,
     Protocol,
 )
 

mev_inspect/classifiers/specs/balancer.py

@@ -1,6 +1,6 @@
+from mev_inspect.schemas.classifier_specs import ClassifierSpec
 from mev_inspect.schemas.classified_traces import (
     Classification,
-    ClassifierSpec,
     Protocol,
 )
 

mev_inspect/classifiers/specs/curve.py

@@ -1,5 +1,5 @@
+from mev_inspect.schemas.classifier_specs import ClassifierSpec
 from mev_inspect.schemas.classified_traces import (
-    ClassifierSpec,
     Protocol,
 )
 

mev_inspect/classifiers/specs/erc20.py

@@ -1,6 +1,6 @@
+from mev_inspect.schemas.classifier_specs import ClassifierSpec
 from mev_inspect.schemas.classified_traces import (
     Classification,
-    ClassifierSpec,
 )
 
 

mev_inspect/classifiers/specs/uniswap.py

@@ -1,6 +1,6 @@
+from mev_inspect.schemas.classifier_specs import ClassifierSpec
 from mev_inspect.schemas.classified_traces import (
     Classification,
-    ClassifierSpec,
     Protocol,
 )
 

mev_inspect/classifiers/specs/weth.py

@@ -1,6 +1,6 @@
+from mev_inspect.schemas.classifier_specs import ClassifierSpec
 from mev_inspect.schemas.classified_traces import (
     Classification,
-    ClassifierSpec,
     Protocol,
 )
 

mev_inspect/classifiers/specs/zero_ex.py

@@ -1,7 +1,5 @@
-from mev_inspect.schemas.classified_traces import (
+from mev_inspect.schemas.classifier_specs import ClassifierSpec
-    ClassifierSpec,
+from mev_inspect.schemas.classified_traces import Protocol
-    Protocol,
-)
 
 
 ZEROX_CONTRACT_SPECS = [

mev_inspect/schemas/classified_traces.py

@@ -1,8 +1,6 @@
 from enum import Enum
 from typing import Any, Dict, List, Optional
 
-from pydantic import BaseModel
-
 from .blocks import Trace
 
 
@@ -64,10 +62,3 @@ class DecodedCallTrace(CallTrace):
     gas_used: Optional[int]
     function_name: Optional[str]
     function_signature: Optional[str]
-
-
-class ClassifierSpec(BaseModel):
-    abi_name: str
-    protocol: Optional[Protocol] = None
-    valid_contract_addresses: Optional[List[str]] = None
-    classifications: Dict[str, Classification] = {}

mev_inspect/schemas/classifier_specs.py

@@ -0,0 +1,12 @@
+from typing import Dict, List, Optional
+
+from pydantic import BaseModel
+
+from .classified_traces import Classification, Protocol
+
+
+class ClassifierSpec(BaseModel):
+    abi_name: str
+    protocol: Optional[Protocol] = None
+    valid_contract_addresses: Optional[List[str]] = None
+    classifications: Dict[str, Classification] = {}