From dfa8abb9e6eea664a3b8bff3d0e6910e27ab37d7 Mon Sep 17 00:00:00 2001 From: Kseniia Astakhova Date: Fri, 29 Dec 2023 22:59:49 +0500 Subject: [PATCH] fix --- .gitignore | 1 + type_inference/type_retrieval_service.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/.gitignore b/.gitignore index b66c2e0..92c08ae 100644 --- 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 index 0c4d669..0d72f39 100644 --- 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}