fix: cleanup

ApeWorX · Aug 29, 2024 · 30ffb64 · 30ffb64
1 parent 6d05c26
commit 30ffb64
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/ape_vyper/compiler.py b/ape_vyper/compiler.py
@@ -994,6 +994,9 @@ def compile_code(
                 # Not sure when this would happen.
                 raise VyperCompileError("Failed to produce contract type.")
 
+            # Clean-up (just in case)
+            file.unlink(missing_ok=True)
+
             return contract_type
 
     def _source_vyper_version(self, code: str) -> Version:

diff --git a/tests/test_compiler.py b/tests/test_compiler.py
@@ -566,7 +566,7 @@ def test_enrich_error_handle_when_name(compiler, geth_provider, mocker):
 
     tb = mocker.MagicMock()
     tb.revert_type = "NONPAYABLE_CHECK"
-    error = ContractLogicError("", source_traceback=tb)
+    error = ContractLogicError(None, source_traceback=tb)
     new_error = compiler.enrich_error(error)
     assert isinstance(new_error, NonPayableError)