diff --git a/services/forecasting/app/services/forecasting_service.py b/services/forecasting/app/services/forecasting_service.py index e765b530..5243a86b 100644 --- a/services/forecasting/app/services/forecasting_service.py +++ b/services/forecasting/app/services/forecasting_service.py @@ -251,7 +251,7 @@ class ForecastingService: # Pass the product_name to the model client model_data = await self.model_client.get_best_model_for_forecasting( tenant_id=tenant_id, - product_name=product_name # Make sure to pass product_name + product_id=product_name # Make sure to pass product_name ) return model_data except Exception as e: diff --git a/services/forecasting/app/services/model_client.py b/services/forecasting/app/services/model_client.py index 906a3d12..df102a3d 100644 --- a/services/forecasting/app/services/model_client.py +++ b/services/forecasting/app/services/model_client.py @@ -66,7 +66,6 @@ class ModelClient: # Get latest model latest_model = await self.clients.training.get_active_model_for_product( tenant_id=tenant_id, - model_type="forecasting", product_name=product_id )