From 6c8501ea2b016f436fb0aba4a9603368fe1ef245 Mon Sep 17 00:00:00 2001
From: davebshow <davebshow@gmail.com>
Date: Sat, 2 Jul 2016 22:05:08 -0400
Subject: [PATCH] cleanup
---
goblin/api.py | 2 +-
goblin/mapper.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/goblin/api.py b/goblin/api.py
index 91a1b37..43d3ba4 100644
--- a/goblin/api.py
+++ b/goblin/api.py
@@ -212,7 +212,7 @@ class ElementMeta(type):
v = properties.PropertyDescriptor(v.data_type)
new_namespace[k] = v
new_namespace['__mapping__'] = mapper.create_mapping(namespace,
- props)
+ props)
result = type.__new__(cls, name, bases, new_namespace)
return result
diff --git a/goblin/mapper.py b/goblin/mapper.py
index ac3bfaf..bf54b2c 100644
--- a/goblin/mapper.py
+++ b/goblin/mapper.py
@@ -48,7 +48,7 @@ class VertexMapping:
def __init__(self, namespace, properties):
self._label = namespace.get('__label__', None)
- self._type = 'vertex'
+ self._type = namespace['__type__']
self._properties = []
self._map_properties(properties)
--
GitLab