diff --git a/goblin/gremlin_python_driver/driver.py b/goblin/gremlin_python_driver/driver.py index dc8b206dd65612cfe6b4cfd4f01d980ca4c822d8..d7467db95d439f4211c38525e39fcf4d73815c9a 100644 --- a/goblin/gremlin_python_driver/driver.py +++ b/goblin/gremlin_python_driver/driver.py @@ -48,11 +48,11 @@ class Driver: class AsyncResponseIter: - def __init__(self, ws, loop, conn, *, force_close=True): + def __init__(self, ws, loop, conn): self._ws = ws self._loop = loop self._conn = conn - self._force_close = force_close + self._force_close = self._conn.force_close self._closed = False async def __aiter__(self): @@ -89,9 +89,14 @@ class AsyncResponseIter: class Connection: - def __init__(self, ws, loop): + def __init__(self, ws, loop, *, force_close=True): self._ws = ws self._loop = loop + self._force_close = force_close + + @property + def force_close(self): + return self._force_close def submit(self, gremlin,