Support latest gds alpha release

Author: FlorentinD

.github/workflows/gds-integration-tests.yml

@@ -20,7 +20,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest]
-        gds-version: ["1.22", "2.0.0a1"]
+        gds-version: ["1.22", "2.0.0a4"]
     defaults:
       run:
         working-directory: python-wrapper

python-wrapper/tests/neo4j_and_gds/gds_helper.py

@@ -4,12 +4,19 @@
 
 from graphdatascience import GdsSessions, GraphDataScience
 from graphdatascience.arrow_client.arrow_authentication import UsernamePasswordAuthentication
-from graphdatascience.semantic_version.semantic_version import SemanticVersion
 from graphdatascience.session import AuraAPICredentials, AuraGraphDataScience, DbmsConnectionInfo, SessionMemory
 from graphdatascience.session.aura_api import AuraApi
 from graphdatascience.session.aura_api_responses import InstanceCreateDetails
 from graphdatascience.version import __version__
 
+# The GDS client >= 2.0 moved `SemanticVersion` under `versions`
+if int(__version__.split(".")[0]) >= 2:
+    from graphdatascience.versions.semantic_version import SemanticVersion
+else:
+    from graphdatascience.semantic_version.semantic_version import (  # type: ignore[import-not-found,no-redef]
+        SemanticVersion,
+    )
+
 
 def parse_version(version: str) -> SemanticVersion:
     server_version_match = re.search(r"(\d+\.)?(\d+\.)?(\*|\d+)", version)

python-wrapper/tests/neo4j_and_gds/test_notebooks.py

@@ -3,8 +3,16 @@
 import pytest
 from dotenv import load_dotenv
 from graphdatascience import GraphDataScience
-from graphdatascience.semantic_version.semantic_version import SemanticVersion
 from graphdatascience.session import AuraGraphDataScience
+from graphdatascience.version import __version__
+
+# The GDS client >= 2.0 moved `SemanticVersion` under `versions`
+if int(__version__.split(".")[0]) >= 2:
+    from graphdatascience.versions.semantic_version import SemanticVersion
+else:
+    from graphdatascience.semantic_version.semantic_version import (  # type: ignore[import-not-found,no-redef]
+        SemanticVersion,
+    )
 
 from tests.neo4j_and_gds.gds_helper import GDS_VERSION
 from tests.notebook_runner import run_notebooks