remove sales value from forecast response
This commit is contained in:
@@ -202,6 +202,7 @@ def _process_sales(
|
||||
df_cust = df_cust.sort_values(by=DATE_FEAT).reset_index()
|
||||
|
||||
# check data availability
|
||||
# TODO rework criteria
|
||||
if len(df_cust) < min_num_data_points:
|
||||
pipe.fail(STATUS_HANDLER.pipe_states.TOO_FEW_POINTS)
|
||||
return pipe
|
||||
@@ -238,6 +239,7 @@ def _process_sales(
|
||||
reg.fit(X_train, y_train, eval_set=[(X_train, y_train), (X_test, y_test)], verbose=100)
|
||||
|
||||
test.loc[:, "vorhersage"] = reg.predict(X_test)
|
||||
test = test.drop(SALES_FEAT, axis=1)
|
||||
test = test.reset_index(drop=True)
|
||||
|
||||
pipe.success(test, STATUS_HANDLER.SUCCESS)
|
||||
@@ -320,3 +322,12 @@ def pipeline_sales(
|
||||
assert pipe.results is not None, "needed export response not set in pipeline"
|
||||
|
||||
return pipe.results
|
||||
|
||||
|
||||
def pipeline_sales_dummy(
|
||||
session: Session,
|
||||
company_id: int | None = None,
|
||||
start_date: Datetime | None = None,
|
||||
) -> SalesPrognosisResultsExport:
|
||||
"""prototype dummy function for tests by DelBar"""
|
||||
...
|
||||
|
||||
Reference in New Issue
Block a user