Skip to content
Snippets Groups Projects
Commit d0745e28 authored by benoit-cty's avatar benoit-cty
Browse files

Fix style

parent 9909f6a0
No related branches found
No related tags found
1 merge request!77PLF 2023
Pipeline #6116 passed
......@@ -103,7 +103,7 @@ def load_data(filename: Optional[str]):
print(f"Loading from file {path} or from table {filename}")
if filename[-3:] == ".h5" or filename[-4:] == ".hdf":
return pandas.read_hdf(path)
elif ".csv" in filename:
elif ".csv" in filename or ".zip" in filename:
return pandas.read_csv(path)
else:
return from_postgres(filename)
......@@ -302,6 +302,7 @@ def dataframe_pondere(dictionnaire_simulations: Dict) -> pandas.DataFrame:
def compare(period: str, dictionnaire_simulations, compute_deciles=True):
logging.debug("Compare.")
res: Total = {}
if (
"avant" not in dictionnaire_simulations
......@@ -330,6 +331,7 @@ def compare(period: str, dictionnaire_simulations, compute_deciles=True):
).sum()
total: Total = res
if compute_deciles:
logging.debug("Calcule des déciles.")
# On rajoute les simulations par défaut à la liste des colonnes sur lesquelles calculer les déciles,
# On en a besoin si ces colonnes ne sont pas déjà dans le dictionnaire_simulations (par exemple
# dans le cas d'un compare avec isdecile = True)
......@@ -421,7 +423,7 @@ def adjustment_new(empiric: dict, annee: int, brute_result: dict):
"""
Facteur d'ajustement en fonction des valeurs par année présentent dans le PLF
"""
logging.debug(f"annee={annee} empiric = {empiric}, brute_result={brute_result}")
logging.debug(f"adjustment_new - annee={annee} empiric = {empiric}, brute_result={brute_result}")
factor_avant = empiric.get(annee) / brute_result["avant"]
if brute_result.get("plf"):
factor_plf = empiric.get(str(int(annee) + 1)) / brute_result["plf"]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment