add meta class to schemas (#95)

Signed-off-by: Allison Suarez Miranda <asuarezmiranda@lyft.com>
amundsen-io · Feb 26, 2021 · a954f1b · a954f1b
1 parent d87597a
commit a954f1b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/amundsen_common/models/lineage.py b/amundsen_common/models/lineage.py
@@ -19,8 +19,9 @@ class LineageItem:
 
 
 class LineageItemSchema(AttrsSchema):
-    target = LineageItem
-    register_as_scheme = True
+    class Meta:
+        target = LineageItem
+        register_as_scheme = True
 
 
 @attr.s(auto_attribs=True, kw_only=True)
@@ -33,5 +34,6 @@ class Lineage:
 
 
 class LineageSchema(AttrsSchema):
-    target = Lineage
-    register_as_scheme = True
+    class Meta:
+        target = Lineage
+        register_as_scheme = True
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 
 from setuptools import find_packages, setup
 
-__version__ = '0.8.0'
+__version__ = '0.8.1'
 
 setup(
     name='amundsen-common',