From c68e6f54bdd92fded2ef1f1754a60ae3a3b24c9b Mon Sep 17 00:00:00 2001 From: mmaurostoffel <166130318+mmaurostoffel@users.noreply.github.com> Date: Sun, 5 Jan 2025 13:19:43 +0100 Subject: [PATCH] cleanup commit --- etl/src/api/main.py | 3 ++- etl/src/data/etl_region_capacities.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/etl/src/api/main.py b/etl/src/api/main.py index 81a8571..226a515 100644 --- a/etl/src/api/main.py +++ b/etl/src/api/main.py @@ -1,6 +1,7 @@ import data import polars as pl from data import etl_property_capacities as etl_pc +from data import etl_region_capacities as etl_rc from fastapi import FastAPI, Response d = data.load() @@ -46,7 +47,7 @@ def property_base_data(id: int): @app.get("/region/{id}/capacities") def region_capacities_data(id: int): - capacities = etl_pc.region_capacities(id) + capacities = etl_rc.region_capacities(id) return capacities diff --git a/etl/src/data/etl_region_capacities.py b/etl/src/data/etl_region_capacities.py index 3d65f3f..f4534cc 100644 --- a/etl/src/data/etl_region_capacities.py +++ b/etl/src/data/etl_region_capacities.py @@ -1,4 +1,4 @@ -from etl.src import data +import data import polars as pl from io import StringIO import numpy as np