railjson-generator: fix example script generation dir

Author: Baptiste Prevot

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"))