From a496d360e8b2ba4f12875e8c8e42f9b14448db17 Mon Sep 17 00:00:00 2001 From: Baptiste Prevot Date: Tue, 6 Jun 2023 10:37:54 +0200 Subject: [PATCH] railjson-generator: fix example script generation dir --- .../railjson_generator/scripts/examples/example_script.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/python/railjson_generator/railjson_generator/scripts/examples/example_script.py b/python/railjson_generator/railjson_generator/scripts/examples/example_script.py index 4768929cd63..98520adbec7 100755 --- a/python/railjson_generator/railjson_generator/scripts/examples/example_script.py +++ b/python/railjson_generator/railjson_generator/scripts/examples/example_script.py @@ -2,10 +2,10 @@ from pathlib import Path -from railjson_generator import InfraBuilder, Location, SimulationBuilder +from railjson_generator import InfraBuilder, Location, SimulationBuilder, get_output_dir from railjson_generator.schema.infra.direction import Direction -CURRENT_DIR = Path(__file__).parent +OUTPUT_DIR = get_output_dir() # GENERATE INFRA builder = InfraBuilder() @@ -60,7 +60,7 @@ infra = builder.build() # Save railjson -infra.save(CURRENT_DIR / Path("infra.json")) +infra.save(OUTPUT_DIR / Path("infra.json")) # GENERATE SIMULATION builder = SimulationBuilder() @@ -86,4 +86,4 @@ sim = builder.build() # Save railjson -sim.save(CURRENT_DIR / Path("simulation.json")) +sim.save(OUTPUT_DIR / Path("simulation.json"))