From 89aa08cce7f33e6dafced6f8619f1fc4afdc3c91 Mon Sep 17 00:00:00 2001 From: davebshow <davebshow@gmail.com> Date: Tue, 12 Jul 2016 16:02:34 -0400 Subject: [PATCH] simplified function logic --- goblin/traversal.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/goblin/traversal.py b/goblin/traversal.py index 0c2dba3..0f326ea 100644 --- a/goblin/traversal.py +++ b/goblin/traversal.py @@ -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, -- GitLab