fix: timeout issues in gremlinpython==3.4.9 (#17)

* Fix timeout issues in tornado

Signed-off-by: Rajat Jaiswal <rajat.jaiswal@hotstar.com>

* Bump the version

Signed-off-by: Rajat Jaiswal <rajat.jaiswal@hotstar.com>

amundsen_gremlin/neptune_bulk_loader/api.py

@@ -22,6 +22,7 @@
 from gremlin_python.driver.driver_remote_connection import (
     DriverRemoteConnection
 )
+from gremlin_python.driver.tornado.transport import TornadoTransport
 from gremlin_python.process.anonymous_traversal import traversal
 from gremlin_python.process.graph_traversal import GraphTraversalSource
 from neptune_python_utils.endpoints import Endpoints, RequestParameters
@@ -87,7 +88,7 @@ def create_graph_traversal_source(**kwargs: Any) -> GraphTraversalSource:
         prepared_request = override_prepared_request_parameters(
             endpoints.gremlin_endpoint().prepare_request(), override_uri=override_uri)
         kwargs['traversal_source'] = 'g'
-        remote_connection = DriverRemoteConnection(url=prepared_request, **kwargs)
+        remote_connection = DriverRemoteConnection(url=prepared_request, transport_factory=lambda: TornadoTransport(read_timeout=None, write_timeout=None), **kwargs)
         return traversal().withRemote(remote_connection)
     return create_graph_traversal_source
 

setup.py

@@ -10,7 +10,7 @@
 
 setup(
     name='amundsen-gremlin',
-    version='0.0.9',
+    version='0.0.10',
     description='Gremlin code library for Amundsen',
     url='https://github.com/amundsen-io/amundsengremlin',
     maintainer='Amundsen TSC',