bugfix: Fix Ammonite tests

scalameta · Oct 22, 2024 · 3e1223c · 3e1223c
1 parent 48a63d1
commit 3e1223c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metals/src/main/scala/scala/meta/internal/parsing/Trees.scala b/metals/src/main/scala/scala/meta/internal/parsing/Trees.scala
@@ -174,7 +174,7 @@ final class Trees(
       val skipFistShebang =
         if (text.startsWith("#!")) text.replaceFirst("#!", "//") else text
       val input = Input.VirtualFile(path.toURI.toString(), skipFistShebang)
-      if (path.isAmmoniteScript) {
+      if (path.isAmmoniteScript || path.isMill) {
         val ammoniteInput = Input.Ammonite(input)
         ammoniteInput.safeParse[MultiSource](dialect)
       } else {