This is an automated email from the ASF dual-hosted git repository.

skrawcz pushed a commit to branch stefan/fix-unit-tests
in repository https://gitbox.apache.org/repos/asf/hamilton.git

commit 44ab7693d2e29eba0f7e3e0ace70499d333274d4
Author: Stefan Krawczyk <[email protected]>
AuthorDate: Mon Dec 29 22:20:56 2025 +1100

    Fixes LSP tests
---
 dev_tools/language_server/hamilton_lsp/server.py | 2 +-
 dev_tools/language_server/tests/ls_setup.py      | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/dev_tools/language_server/hamilton_lsp/server.py 
b/dev_tools/language_server/hamilton_lsp/server.py
index f09254b3..3be65a94 100644
--- a/dev_tools/language_server/hamilton_lsp/server.py
+++ b/dev_tools/language_server/hamilton_lsp/server.py
@@ -39,7 +39,7 @@ from lsprotocol.types import (
     SymbolKind,
     VersionedTextDocumentIdentifier,
 )
-from pygls.server import LanguageServer
+from pygls.lsp.server import LanguageServer
 
 from hamilton import ad_hoc_utils
 from hamilton.graph import FunctionGraph, create_graphviz_graph
diff --git a/dev_tools/language_server/tests/ls_setup.py 
b/dev_tools/language_server/tests/ls_setup.py
index f2ea066e..8bfa86b0 100644
--- a/dev_tools/language_server/tests/ls_setup.py
+++ b/dev_tools/language_server/tests/ls_setup.py
@@ -21,7 +21,7 @@ import threading
 
 import pytest
 from lsprotocol.types import EXIT, INITIALIZE, SHUTDOWN, ClientCapabilities, 
InitializeParams
-from pygls.server import LanguageServer
+from pygls.lsp.server import LanguageServer
 
 RETRIES = 3
 CALL_TIMEOUT = 3

Reply via email to