fix

EvgSkv · Dec 29, 2023 · dfa8abb · dfa8abb
1 parent 5d23f43
commit dfa8abb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -4,3 +4,4 @@
 .idea
 __pycache__/
 docs/.DS_Store
+env
diff --git a/type_inference/type_retrieval_service.py b/type_inference/type_retrieval_service.py
@@ -136,7 +136,7 @@ def RetrieveTypes(self, filename):
 GROUP BY table_name
 HAVING table_name IN UNNEST(@tables);''', job_config)
     data_by_table_name = query.to_dataframe().set_index('table_name')
-    columns_by_table_name = data_by_table_name.to_dict()['columns']
+    columns_by_table_name = data_by_table_name.to_dict()['columns'].items()
     columns = {table: json.loads(type)
                for table, type in columns_by_table_name}
-Original file line number
+Diff line change
@@ Expand Up / @@ -4,3 +4,4 @@ @@
     .idea
     __pycache__/
     docs/.DS_Store
+    env