From b229794e6d87244443809ccad15b77343b655e73 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Florian=20F=C3=B6rster?= Date: Wed, 19 Feb 2025 12:28:11 +0100 Subject: [PATCH] change param to customer ID --- src/delta_barth/analysis/forecast.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/delta_barth/analysis/forecast.py b/src/delta_barth/analysis/forecast.py index a9b54a5..26509f7 100644 --- a/src/delta_barth/analysis/forecast.py +++ b/src/delta_barth/analysis/forecast.py @@ -34,7 +34,7 @@ if TYPE_CHECKING: def sales_per_customer( df: pd.DataFrame, - kunde: int, + customer_id: int, min_num_data_points: int = 100, ) -> FcResult: """_summary_ @@ -59,7 +59,9 @@ def sales_per_customer( # TODO change away from nested DataFrames: just use "f_umsatz_fakt" # TODO with strong type checks df = df.copy() - df_firma = df[(df["firma_refid"] == kunde) & (df["beleg_typ"] == 1) & (df["betrag"] > 0)] + df_firma = df[ + (df["firma_refid"] == customer_id) & (df["beleg_typ"] == 1) & (df["betrag"] > 0) + ] for transaction in df_firma["vorgang_refid"].unique(): cust_data.order.append(transaction)