Use Union instead of | for typing

due to python 3.9

Author: FlorentinD

graphdatascience/arrow_client/v2/api_types.py

@@ -1,3 +1,4 @@
+from typing import Optional
 from graphdatascience.arrow_client.arrow_base_model import ArrowBaseModel
 
 
@@ -19,15 +20,15 @@ def progress_known(self) -> bool:
             return False
         return True
 
-    def progress_percent(self) -> float | None:
+    def progress_percent(self) -> Optional[float]:
         if self.progress_known():
             return self.progress * 100
         return None
 
     def base_task(self) -> str:
         return self.description.split("::")[0].strip()
 
-    def sub_tasks(self) -> str | None:
+    def sub_tasks(self) -> Optional[str]:
         task_split = self.description.split("::", maxsplit=1)
         if len(task_split) > 1:
             return task_split[1].strip()

graphdatascience/arrow_client/v2/job_client.py

@@ -14,7 +14,7 @@
 
 
 class JobClient:
-    def __init__(self, progress_bar_options: dict[str, Any] | None = None):
+    def __init__(self, progress_bar_options: Optional[dict[str, Any]] = None):
         self._progress_bar_options = progress_bar_options or {}
 
     @staticmethod

graphdatascience/query_runner/protocol/write_protocols.py

@@ -127,7 +127,7 @@ def run_write_back(
 
 
 class RemoteWriteBackV3(WriteProtocol):
-    def __init__(self, progress_bar_options: dict[str, Any] | None = None):
+    def __init__(self, progress_bar_options: Optional[dict[str, Any]] = None):
         self._progress_bar_options = progress_bar_options or {}
 
     def write_back_params(

graphdatascience/session/aura_api_responses.py

@@ -32,7 +32,7 @@ class SessionDetails:
     def from_json(cls, data: dict[str, Any]) -> SessionDetails:
         id = data["id"]
         expiry_date = data.get("expiry_date")
-        ttl: Any | None = data.get("ttl")
+        ttl: Optional[Any] = data.get("ttl")
         instance_id = data.get("instance_id")
         cloud_location = CloudLocation(data["cloud_provider"], data["region"]) if data.get("cloud_provider") else None
 
@@ -65,7 +65,7 @@ def from_json_with_error(cls, data: dict[str, Any], errors: list[dict[str, Any]]
 
         id = data["id"]
         expiry_date = data.get("expiry_date")
-        ttl: Any | None = data.get("ttl")
+        ttl: Optional[Any] = data.get("ttl")
         instance_id = data.get("instance_id")
         cloud_location = CloudLocation(data["cloud_provider"], data["region"]) if data.get("cloud_provider") else None
 

graphdatascience/session/session_info.py

@@ -2,7 +2,7 @@
 
 from dataclasses import dataclass
 from datetime import datetime, timedelta
-from typing import Optional
+from typing import Optional, Union
 
 from graphdatascience.session.aura_api_responses import SessionDetails, SessionDetailsWithErrors, SessionErrorData
 from graphdatascience.session.cloud_location import CloudLocation
@@ -41,7 +41,7 @@ class SessionInfo:
     errors: Optional[list[SessionErrorData]] = None
 
     @classmethod
-    def from_session_details(cls, details: SessionDetailsWithErrors | SessionDetails) -> SessionInfo:
+    def from_session_details(cls, details: Union[SessionDetailsWithErrors, SessionDetails]) -> SessionInfo:
         errors: Optional[list[SessionErrorData]] = None
         if isinstance(details, SessionDetailsWithErrors):
             errors = details.errors