diff --git a/src/components/preprocessing/base_kpi_inference_curator.py b/src/components/preprocessing/base_kpi_inference_curator.py index 1a2e2ae..e1c73ea 100644 --- a/src/components/preprocessing/base_kpi_inference_curator.py +++ b/src/components/preprocessing/base_kpi_inference_curator.py @@ -197,7 +197,7 @@ def return_sliced_squad(self, squad_json, indices): return {} pdf2pars = defaultdict(list) - for (i1, i2) in indices: + for i1, i2 in indices: pdf2pars[i1].append(i2) data = [] diff --git a/src/components/preprocessing/kpi_inference_curator.py b/src/components/preprocessing/kpi_inference_curator.py index 4b4b72f..b9da5af 100644 --- a/src/components/preprocessing/kpi_inference_curator.py +++ b/src/components/preprocessing/kpi_inference_curator.py @@ -103,6 +103,7 @@ def clean(self, df): Args: df (A pandas dataframe) """ + # map kpi to question def map_kpi(r): try: