Trace naming suggestions

mev_inspect/classifiers/trace.py

@@ -6,8 +6,8 @@ from mev_inspect.schemas.blocks import CallAction, CallResult, Trace, TraceType
 from mev_inspect.schemas.classified_traces import (
     Classification,
     ClassifiedTrace,
-    Call,
-    ClassifiedCall,
+    CallTrace,
+    ClassifiedCallTrace,
     ClassifierSpec,
 )
 
@@ -71,7 +71,7 @@ class TraceClassifier:
                     signature, Classification.unknown
                 )
 
-                return ClassifiedCall(
+                return ClassifiedCallTrace(
                     **trace.dict(),
                     trace_type=trace.type,
                     classification=classification,
@@ -87,7 +87,7 @@ class TraceClassifier:
                     gas_used=result.gas_used if result is not None else None,
                 )
 
-        return Call(
+        return CallTrace(
             **trace.dict(),
             trace_type=trace.type,
             classification=Classification.unknown,

mev_inspect/schemas/classified_traces.py

@@ -28,7 +28,7 @@ class Protocol(Enum):
     curve = "curve"
 
 
-# -------- Trace Models ------------------------------------------------------------------------------
+# -------- Traces ------------------------------------------------------------------------------
 
 
 class ClassifiedTrace(Trace):
@@ -52,13 +52,13 @@ class ClassifiedTrace(Trace):
     abi_name: Optional[str]
 
 
-class Call(ClassifiedTrace):
+class CallTrace(ClassifiedTrace):
 
     to_address: str
     from_address: str
 
 
-class ClassifiedCall(Call):
+class ClassifiedCallTrace(CallTrace):
 
     inputs: Dict[str, Any]
     abi_name: str

tests/helpers.py

@@ -4,8 +4,8 @@ from mev_inspect.schemas.blocks import TraceType
 from mev_inspect.schemas.classified_traces import (
     Classification,
     ClassifiedTrace,
-    Call,
-    ClassifiedCall,
+    CallTrace,
+    ClassifiedCallTrace,
 )
 
 
@@ -18,7 +18,7 @@ def make_transfer_trace(
     token_address: str,
     amount: int,
 ):
-    return Call(
+    return CallTrace(
         transaction_hash=transaction_hash,
         block_number=block_number,
         type=TraceType.call,
@@ -43,7 +43,7 @@ def make_swap_trace(
     recipient_address: str,
     recipient_input_key: str,
 ):
-    return ClassifiedCall(
+    return ClassifiedCallTrace(
         transaction_hash=transaction_hash,
         block_number=block_number,
         type=TraceType.call,