Commit 89aa08cc authored by davebshow's avatar davebshow
Browse files

simplified function logic

parent 9b7e69a7
......@@ -66,6 +66,7 @@ class Traversal(connection.AbstractConnection):
return self._session
def traversal(self):
traversal = self.graph.traversal()
if self._element_class:
label = self._element_class.__mapping__.label
traversal = self._graph.traversal()
......@@ -74,8 +75,6 @@ class Traversal(connection.AbstractConnection):
if self._element_class.__type__ == 'edge':
traversal = traversal.E()
traversal = traversal.hasLabel(label)
else:
traversal = self.graph.traversal()
return traversal
async def submit(self,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment