Merge pull request #879 from basedosdados/staging/fix_cvm

[Bugfix] br_cvm_fii
basedosdados · Oct 31, 2024 · 2a9c3a5 · 2a9c3a5
2 parents 7c52602 + 5ddca88
commit 2a9c3a5
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pipelines/datasets/br_cvm_fi/tasks.py b/pipelines/datasets/br_cvm_fi/tasks.py
@@ -225,8 +225,13 @@ def clean_data_and_make_partitions(path: str, table_id: str) -> str:
     for file in files:
         df = pd.read_csv(f"{path}{file}", sep=";")
         log(f"File {file} read.")
+
+        df.rename(columns={'CNPJ_FUNDO_CLASSE':'CNPJ_FUNDO'}, inplace=True)
+
         df["CNPJ_FUNDO"] = df["CNPJ_FUNDO"].str.replace(r"[/.-]", "")
+
         df = rename_columns(df_arq, df)
+
         df = check_and_create_column(
             df, colunas_totais=cvm_constants.COLUNAS_FINAL_INF.value
         )