fix(gql.transport.websockets): update type to permit passing floats

Author: ossareh

gql/transport/websockets.py

@@ -92,10 +92,10 @@ def __init__(
         headers: Optional[HeadersLike] = None,
         ssl: Union[SSLContext, bool] = False,
         init_payload: Dict[str, Any] = {},
-        connect_timeout: int = 10,
-        close_timeout: int = 10,
-        ack_timeout: int = 10,
-        keep_alive_timeout: Optional[int] = None,
+        connect_timeout: Optional[Union[int, float]] = 10,
+        close_timeout: Optional[Union[int, float]] = 10,
+        ack_timeout: Optional[Union[int, float]] = 10,
+        keep_alive_timeout: Optional[Union[int, float]] = None,
         connect_args: Dict[str, Any] = {},
     ) -> None:
         """Initialize the transport with the given parameters.
@@ -105,10 +105,11 @@ def __init__(
         :param ssl: ssl_context of the connection. Use ssl=False to disable encryption
         :param init_payload: Dict of the payload sent in the connection_init message.
         :param connect_timeout: Timeout in seconds for the establishment
-            of the websocket connection.
-        :param close_timeout: Timeout in seconds for the close.
+            of the websocket connection. If None is provided this will wait forever.
+        :param close_timeout: Timeout in seconds for the close. If None is provided
+            this will wait forever.
         :param ack_timeout: Timeout in seconds to wait for the connection_ack message
-            from the server.
+            from the server. If None is provided this will wait forever.
         :param keep_alive_timeout: Optional Timeout in seconds to receive
             a sign of liveness from the server.
         :param connect_args: Other parameters forwarded to websockets.connect
@@ -118,9 +119,9 @@ def __init__(
         self.headers: Optional[HeadersLike] = headers
         self.init_payload: Dict[str, Any] = init_payload
 
-        self.connect_timeout: int = connect_timeout
-        self.close_timeout: int = close_timeout
-        self.ack_timeout: int = ack_timeout
+        self.connect_timeout: Optional[Union[int, float]] = connect_timeout
+        self.close_timeout: Optional[Union[int, float]] = close_timeout
+        self.ack_timeout: Optional[Union[int, float]] = ack_timeout
         self.keep_alive_timeout: Optional[int] = keep_alive_timeout
 
         self.connect_args = connect_args