From 8fae7851a634017ac55bc51882eb941e742f3b01 Mon Sep 17 00:00:00 2001 From: Daniel Rosel Date: Fri, 28 Nov 2025 17:38:25 +0100 Subject: [PATCH] migrating pricers --- experiments/procesing/steps/pricing.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/experiments/procesing/steps/pricing.py b/experiments/procesing/steps/pricing.py index 95f18e3..f8a8eb3 100755 --- a/experiments/procesing/steps/pricing.py +++ b/experiments/procesing/steps/pricing.py @@ -1,7 +1,7 @@ import numpy as np import pandas as pd from procesing.steps.base import BaseContextStep -from procesing.pricing import ElasticityBasedPricingFunction +from procesing.pricers import ElasticityBasedPricer class StateSpace: """State representation for pricing functions""" @@ -57,7 +57,7 @@ class FitPricingFunctionStep(BaseContextStep): """ def transform(self, elasticity_df: pd.DataFrame): - pricing_class = self.context.config.get('pricing_function_class', ElasticityBasedPricingFunction) + pricing_class = self.context.config.get('pricing_function_class', ElasticityBasedPricer) pricing_params = self.context.config.get('pricing_function_params', {}) pricer = pricing_class(**pricing_params) @@ -79,7 +79,7 @@ class PredictPricesStep(BaseContextStep): products = self.context.products product_ids = products['id'].values - predicted_prices = pricer.transform(state_space, product_ids) + predicted_prices = pricer.predict(state_space) return pd.DataFrame({ 'productId': product_ids,