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