hard coded buckets

This commit is contained in:
Khalim Conn-Kowlessar 2026-03-27 01:46:58 +00:00
parent 80a5d0252b
commit d287df1b82
2 changed files with 12 additions and 1 deletions

View file

@ -83,7 +83,16 @@ def test_rebaselining_pipeline_with_real_data():
model_api = ModelApi(
portfolio_id="test-portfolio",
timestamp=datetime.now().isoformat(),
prediction_buckets=get_prediction_buckets(),
prediction_buckets={
"sap_change_predictions": "retrofit-sap-predictions-dev",
"heat_demand_predictions": "retrofit-heat-predictions-dev",
"carbon_change_predictions": "retrofit-carbon-predictions-dev",
"heating_kwh_predictions": "retrofit-heating-kwh-predictions-dev",
"hotwater_kwh_predictions": "retrofit-hotwater-kwh-predictions-dev",
"retrofit_sap_baseline_predictions": "retrofit-sap-baseline-predictions-dev",
"retrofit_carbon_baseline_predictions": "retrofit-carbon-baseline-predictions-dev",
"retrofit_heat_baseline_predictions": "retrofit-heat-baseline-predictions-dev",
},
max_retries=1
)
bucket = "retrofit-data-dev"

View file

@ -4,3 +4,5 @@ log_cli = true
log_cli_level = INFO
addopts = --cov-report term-missing --cov=etl/epc --cov=recommendations --cov=backend --cov=etl/epc_clean --cov=etl/spatial
testpaths = recommendations/tests backend/tests etl/epc/tests etl/epc_clean/tests etl/spatial/tests backend/condition/tests backend/address2UPRN/tests backend/onboarders/tests backend/categorisation/tests backend/export/tests
markers =
integration: mark a test as an integration test