From 2baf2b1bf24ecb90c3aea273535be400605cf1dd Mon Sep 17 00:00:00 2001
From: Jesse Hines <hinesjr@ornl.gov>
Date: Thu, 18 Sep 2025 15:32:41 -0400
Subject: [PATCH] Fix multi-part snapshots

---
 raps/engine.py    | 2 +-
 raps/telemetry.py | 6 +++++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/raps/engine.py b/raps/engine.py
index 9ae27b1..8d7e231 100644
--- a/raps/engine.py
+++ b/raps/engine.py
@@ -248,7 +248,7 @@ class Engine:
                 if len(snap_map) > 0:
                     if partition_short not in snap_map:
                         raise RuntimeError(f"Snapshot '{partition_short}.npz' not in {sim_config.replay[0]}")
-                    replay_files = snap_map[partition_short]
+                    replay_files = [snap_map[partition_short]]
                 else:
                     replay_files = sim_config.replay
             else:
diff --git a/raps/telemetry.py b/raps/telemetry.py
index 6d5aa19..915fc97 100644
--- a/raps/telemetry.py
+++ b/raps/telemetry.py
@@ -61,6 +61,10 @@ class TelemetryArgs(RAPSBaseModel):
             raise ValueError("Either --live or --replay is required")
         return self
 
+    @property
+    def system_name(self):
+        return self.system
+
 
 shortcuts = {
     "replay": "f",
@@ -208,7 +212,7 @@ class Telemetry:
             for file in files:
                 print(f"Loading {file}")
                 new_data, args_from_file = self.load_snapshot(file)
-                print(f"File was generated with: --system {args_from_file.system}")
+                print(f"File was generated with: --system {args_from_file.system_name}")
                 if not data:
                     data = new_data
                 else:
-- 
GitLab