added category class for dataset categories and labels

Frater-SDK · Mar 10, 2020 · f11b1be · f11b1be
1 parent 67926a4
commit f11b1be
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git a/frater/category/__init__.py b/frater/category/__init__.py
@@ -0,0 +1 @@
+from .category import Category
diff --git a/frater/category/category.py b/frater/category/category.py
@@ -0,0 +1,10 @@
+from dataclasses import dataclass
+
+from dataclasses_json import DataClassJsonMixin
+
+
+@dataclass
+class Category(DataClassJsonMixin):
+    index: int = 0
+    label: str = ''
+    dataset: str = ''
diff --git a/setup.py b/setup.py
@@ -10,7 +10,7 @@
 
 setup(
     name='frater',
-    version='0.3.1.4',
+    version='0.3.1.5',
     packages=packages,
     license='MIT',
     url='https://github.com/frater-sdk/frater',