Require protocol to build a swap

2022-01-20 19:03:29 -05:00 · 2022-01-20 19:03:29 -05:00 · 3afb854d13
commit 3afb854d13
parent a056919507
1 changed files with 2 additions and 2 deletions
--- a/tests/helpers.py
+++ b/tests/helpers.py
@ -1,4 +1,4 @@
-from typing import List, Optional
+from typing import List

 from mev_inspect.schemas.traces import (
    Classification,
@ -48,7 +48,7 @@ def make_swap_trace(
    contract_address: str,
    abi_name: str,
    function_signature: str,
-    protocol: Optional[Protocol],
+    protocol: Protocol,
    recipient_address: str,
    recipient_input_key: str,
 ):