From ee1e1162161d009aa984601cdbed9f90f43f88d1 Mon Sep 17 00:00:00 2001
From: davebshow <davebshow@gmail.com>
Date: Thu, 8 Jun 2017 18:08:33 -0400
Subject: [PATCH] simple refactor cleanup
---
aiogremlin/remote/remote_connection.py | 6 ++----
1 file changed, 2 insertions(+), 4 deletions(-)
diff --git a/aiogremlin/remote/remote_connection.py b/aiogremlin/remote/remote_connection.py
index 9cc8f65..104f94a 100644
--- a/aiogremlin/remote/remote_connection.py
+++ b/aiogremlin/remote/remote_connection.py
@@ -1,9 +1,7 @@
-from gremlin_python.process import traversal
+from gremlin_python.driver.remote_connection import RemoteStrategy
-class AsyncRemoteStrategy(traversal.TraversalStrategy):
- def __init__(self, remote_connection):
- self.remote_connection = remote_connection
+class AsyncRemoteStrategy(RemoteStrategy):
async def apply(self, traversal):
if traversal.traversers is None:
--
GitLab