fix: debug to take into account data chunking

app.py

@@ -1,12 +1,13 @@
 from src.search.ga import GeneticSearch
 from src.hw_nats_fast_interface import HW_NATS_FastInterface
-from src.utils import DEVICES, DATASETS
+from src.utils import DEVICES, DATASETS, union_of_dicts
 import streamlit as st
 import numpy as np
 import pandas as pd
 import matplotlib.pyplot as plt
 import plotly.graph_objects as go
 from collections import OrderedDict
+import json
 st.set_page_config(layout="wide")
 
 TIME_TO_SCORE_EACH_ARCHITECTURE=0.15
@@ -98,8 +99,15 @@ def main():
     # best architecture index
     best_arch_idx = 9930
 
+    nebulos_chunks = []
+    for i in range(4):  # the number of chunks is 4 in this case
+        with open(f"data/nebuloss_{i+1}.json", "r") as f:
+            nebulos_chunks.append(json.load(f))
+    
+    searchspace_dict = union_of_dicts(nebulos_chunks)
+
     # Trigger the search and plot NebulOS Architecture
-    searchspace_interface = HW_NATS_FastInterface(device=device, dataset=dataset)
+    searchspace_interface = HW_NATS_FastInterface(datapath=searchspace_dict, device=device, dataset=dataset)
     search = GeneticSearch(
         searchspace=searchspace_interface,
         fitness_weights=np.array([performance_weight, hardware_weight])