add NAF 2008 code

InseeFrLab · Oct 24, 2024 · 259b43a · 259b43a
1 parent ab60ba9
commit 259b43a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils/extract-train-data-otm.py b/utils/extract-train-data-otm.py
@@ -25,7 +25,7 @@ def transform_json_to_dataframe(json_dir: str):
         libelle = data["task"]["data"]["libelle"]
         liasse_numero = data["task"]["data"]["liasse_numero"]
         date_modification = data["task"]["data"]["date_modification"]
-        NAF2008_code = data["task"]["data"]["apet_finale"]
+        apet_finale = data["task"]["data"]["apet_finale"]
         mode_calcul_ape = data["task"]["data"]["mode_calcul_ape"]
         evenement_type = data["task"]["data"]["evenement_type"]
         liasse_type = data["task"]["data"]["liasse_type"]
@@ -87,7 +87,7 @@ def transform_json_to_dataframe(json_dir: str):
             "cj": cj,
             "date_modification": date_modification,
             "annotation_date": annotation_date,
-            "NAF2008_code": NAF2008_code,
+            "NAF2008_code": apet_finale,
             "mode_calcul_ape": mode_calcul_ape,
             "apet_manual": apet_manual,
             "commentary": commentary,