diff --git a/raps/engine.py b/raps/engine.py index 9ae27b127963e35c5ade824cb7da50a0710d5194..8d7e2310b4708daf464ae3250824bb4ca48fcefb 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 6d5aa19a25bac3c91abca7c6bc66bcb91e683a3d..915fc975a7fc021ced2c95146b046d6566534d9e 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: