add transformation_dict

Author: stantaneous

models.py

@@ -1,7 +1,7 @@
 import pandas as pd
 
 # Lists of Possible entities
-data_transformations = ['Join', 'Union']
+
 join_types = ['left', 'right', 'outer', 'inner']
 
 
@@ -21,4 +21,5 @@ def __init__(self, table, type='user_table'):
 									 'B': ['B0', 'B1', 'B2'],
 									 'id': ['K0', 'K1', 'K3']
 									 }))
-}
+}
+

requirements.txt

@@ -2,4 +2,5 @@ pandas
 numpy
 fuzzywuzzy
 python-Levenshtein
-nltk
+nltk
+sklearn

transformation_functions.py

@@ -24,10 +24,18 @@ def join(table_a, table_b):
 	matched_keys_right = nlp_join_keys_right.get_matches(table_b_columns, threshold=90)
 
 
-	joined_table = pd.merge(table_a, table_b, how=TypeofJoin, left_on=matched_keys_left, right_on=matched_keys_right)
+	joined_table = pd.merge(table_a.table, table_b.table, how=TypeofJoin, left_on=matched_keys_left, right_on=matched_keys_right)
 	return joined_table
 
 
+def union(table_a, table_b):
+	pass
+
+transformations_dict = {
+	'join' : join,
+	'union': union
+}
+
 # --------------------- for debugging ----------------------------------
 
 if __name__ == '__main__':