Skip to content

Commit

Permalink
example: progress bar
Browse files Browse the repository at this point in the history
add example that shows generating config files that displays a progress bar
  • Loading branch information
aaraney committed Mar 25, 2024
1 parent 3b02595 commit 99368d0
Showing 1 changed file with 28 additions and 0 deletions.
28 changes: 28 additions & 0 deletions python/ngen_config_gen/examples/progress_bar/main.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
import geopandas as gpd
import pandas as pd
import tqdm

from ngen.config_gen.file_writer import DefaultFileWriter
from ngen.config_gen.hook_providers import DefaultHookProvider
from ngen.config_gen.generate import generate_configs

from ngen.config_gen.models.cfe import Cfe
from ngen.config_gen.models.pet import Pet

if __name__ == "__main__":
# or pass local file paths instead
hf_file = "https://lynker-spatial.s3.amazonaws.com/v20.1/gpkg/nextgen_09.gpkg"
hf_lnk_file = "https://lynker-spatial.s3.amazonaws.com/v20.1/model_attributes/nextgen_09.parquet"

hf: gpd.GeoDataFrame = gpd.read_file(hf_file, layer="divides")
hf_lnk_data: pd.DataFrame = pd.read_parquet(hf_lnk_file)

hook_provider = DefaultHookProvider(hf=hf, hf_lnk_data=hf_lnk_data)
# files will be written to ./config
file_writer = DefaultFileWriter("./config/")

generate_configs(
hook_providers=tqdm.tqdm(hook_provider),
hook_objects=[Cfe, Pet],
file_writer=file_writer,
)

0 comments on commit 99368d0

Please sign in to comment.