also allow vyper interfaces

foundry-rs · Oct 14, 2024 · 71c6ec7 · 71c6ec7
1 parent 877ab1c
commit 71c6ec7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/crates/compilers/src/project_util/mod.rs b/crates/compilers/src/project_util/mod.rs
@@ -98,7 +98,7 @@ pub(crate) fn create_contract_file(path: &Path, content: impl AsRef<str>) -> Res
 
 fn contract_file_name(name: &str) -> String {
     let name = name.trim();
-    if name.ends_with(".sol") || name.ends_with(".vy") {
+    if name.ends_with(".sol") || name.ends_with(".vy") || name.ends_with(".vyi") {
         name.to_string()
     } else {
         format!("{name}.sol")