Fix oracle loader timeline test fixture

This commit is contained in:
Xin Wang
2026-09-11 10:38:45 +08:00
parent a57e827606
commit 4bfea907b8
+3 -1
View File
@@ -225,7 +225,9 @@ class OracleFeatureTests(unittest.TestCase):
"audio_valid": torch.ones(n, dtype=torch.bool), "audio_valid": torch.ones(n, dtype=torch.bool),
"persons": { "persons": {
"features": torch.randn(n, 6), "features": torch.randn(n, 6),
"frame_times": torch.arange(n, dtype=torch.float64) / 5, # Oracle person rows are current-state observations sampled at
# the same inclusive 5 Hz right edges as the target timeline.
"frame_times": times.clone(),
"frame_offsets": torch.arange(n + 1, dtype=torch.long), "frame_offsets": torch.arange(n + 1, dtype=torch.long),
"records": [{"id": "3"} for _ in range(n)], "records": [{"id": "3"} for _ in range(n)],
}, },