check for mypy.ini too

microsoft · Aug 20, 2024 · 8b68a2c · 8b68a2c
1 parent f60ecd8
commit 8b68a2c
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/bundled/tool/lsp_server.py b/bundled/tool/lsp_server.py
@@ -658,9 +658,12 @@ def get_cwd(settings: Dict[str, Any], document: Optional[workspace.Document]) ->
         # until we leave the workspace
         while candidate.is_relative_to(workspaceFolder):
             # check if pyproject exists
-            if (candidate / "pyproject.toml").is_file():
-                log_to_output(f"found pyproject in {candidate}", lsp.MessageType.Debug)
-                return os.fspath(candidate)
+            check_for = ["project.toml", "mypy.ini"]
+            for n in check_for:
+                candidate_file = candidate / n
+                if candidate_file.is_file():
+                    log_to_output(f"found {n}, using {candidate}", lsp.MessageType.Debug)
+                    return os.fspath(candidate)
             # starting from the current file and working our way up
             else:
                 candidate = candidate.parent