Skip to content

Commit

Permalink
Update extract-train-data-otm.py: fix AGRI
Browse files Browse the repository at this point in the history
  • Loading branch information
TheAIWizard authored Oct 24, 2024
1 parent 259b43a commit 1a560f6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion utils/extract-train-data-otm.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ def transform_json_to_dataframe(json_dir: str):
liasse_numero = data["task"]["data"]["liasse_numero"]
date_modification = data["task"]["data"]["date_modification"]
apet_finale = data["task"]["data"]["apet_finale"]
mode_calcul_ape = data["task"]["data"]["mode_calcul_ape"]
mode_calcul_ape = [v for k, v in data["task"]["data"].items() if k.startswith("mode_calcul_ape") and v != ""][0] # un des jsons contient mode_calcul_apen/apet (l'un ou l'autre)
evenement_type = data["task"]["data"]["evenement_type"]
liasse_type = data["task"]["data"]["liasse_type"]
activ_surf_et = str(data["task"]["data"]["activ_surf_et"])
Expand Down

0 comments on commit 1a560f6

Please sign in to comment.