label thing

jlunder00 · Nov 25, 2024 · 54618db · 54618db
1 parent 89653a1
commit 54618db
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/TMN_DataGen/dataset_generator.py b/TMN_DataGen/dataset_generator.py
@@ -14,8 +14,8 @@ class DatasetGenerator:
     def __init__(self):
         """Initialize dataset generator without config - config provided per method call"""
         self.label_map = {
-            'entails': 1,
-            'contradicts': -1,
+            'entailment': 1,
+            'contradiction': -1,
             'neutral': 0
         }
 

diff --git a/setup.py b/setup.py
@@ -10,7 +10,7 @@
 
 setup(
     name="TMN_DataGen",
-    version='0.6.1',
+    version='0.6.2',
     description="Tree Matching Network Data Generator",
     author="toast",
     packages=find_packages(),