Endpoint working for now, time to work on frontend viz

Author: josephmaa

bridger/logging_utils/cache_entry_database.py

@@ -11,17 +11,6 @@ def __init__(self, cache_entries: list[CacheEntry]):
         print(f"Initializing CacheEntryDatabase with {len(cache_entries)} entries")
         start_time = time.time()
 
-        # Debug info about the entries
-        if cache_entries:
-            print(f"First entry type: {type(cache_entries[0])}")
-            print(f"First entry attributes: {dir(cache_entries[0])}")
-            try:
-                print(
-                    f"First entry state shape: {cache_entries[0].state_representative.shape}"
-                )
-            except Exception as e:
-                print(f"Error getting state shape: {e}")
-
         self.cache_entries = cache_entries
         end_time = time.time()
         print(
@@ -49,8 +38,11 @@ def get_top_n_by_sort_key(self, sort_key: "SortKey", n: int):
         print(f"Getting top {n} entries by {sort_key.key}")
         start_time = time.time()
         try:
-            sorted_entries = self.sort_by_key(sort_key())
-            result = sorted_entries[:n]
+            self.sort_by_key(sort_key)
+            result = [
+                cache_entry.state_representative.tolist()
+                for cache_entry in self.cache_entries[:n].copy()
+            ]
             end_time = time.time()
             print(f"Retrieved top {n} entries in {end_time - start_time:.2f} seconds")
             return result
@@ -94,4 +86,4 @@ def __init__(self):
 
 class SampleCountSortKey(SortKey):
     def __init__(self):
-        super().__init__("sample_count")
+        super().__init__("sampled_count")

bridger/logging_utils/object_log_readers.py

@@ -30,7 +30,6 @@ def read_object_log(log_filepath: str):
     with open(log_filepath, "rb") as f:
         while True:
             try:
-                print(f"This is the type: {type(f)}")
                 buffer = pickle.load(f)
 
                 for element in buffer:
@@ -298,7 +297,6 @@ def __init__(self, dirname: str):
         for entry in _read_object_log(
             os.path.dirname(dirname), log_entry.STATE_NORMALIZED_LOG_ENTRY
         ):
-            print(f"")
             self._states[entry.id] = entry.object
 
         # Store visited states sorted by visit count.

tools/web/sibyl.py

@@ -92,7 +92,7 @@ def go_explore_plot_data():
 
     return {
         "states": [
-            cache_entry.state_representative
+            cache_entry.state_representative.tolist()
             for cache_entry in _CACHE_ENTRY_DATABASE.cache_entries
         ],
         "trajectory_length": _CACHE_ENTRY_DATABASE.get_top_n_by_sort_key(
@@ -104,7 +104,7 @@ def go_explore_plot_data():
         "steps_since_led_to_something_new_reset_count": _CACHE_ENTRY_DATABASE.get_top_n_by_sort_key(
             StepsSinceLedToSomethingNewResetCountSortKey(), n
         ),
-        "sample_count": _CACHE_ENTRY_DATABASE.get_top_n_by_sort_key(
+        "sampled_count": _CACHE_ENTRY_DATABASE.get_top_n_by_sort_key(
             SampleCountSortKey(), n
         ),
         "visit_count": _CACHE_ENTRY_DATABASE.get_top_n_by_sort_key(