fix(create-cart-diagram): wrong dataset filenames

BLSQ · Nov 3, 2024 · 9cfa510 · 9cfa510
1 parent 0d727f1
commit 9cfa510
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/create-cart-diagram/pipeline.py b/create-cart-diagram/pipeline.py
@@ -104,9 +104,9 @@ def load_dataset(
     urban: list[dict] = None
     rural: list[dict] = None
     for f in ds.files:
-        if f.filename == "urban_frame.json":
+        if f.filename == "frame_urban.json":
             urban = json.loads(f.read().decode())
-        if f.filename == "rural_frame.json":
+        if f.filename == "frame_rural.json":
             rural = json.loads(f.read().decode())
 
     if urban is None: