diff --git a/test_easycom_oracle_features.py b/test_easycom_oracle_features.py index feb4778..94f5ee5 100644 --- a/test_easycom_oracle_features.py +++ b/test_easycom_oracle_features.py @@ -225,7 +225,9 @@ class OracleFeatureTests(unittest.TestCase): "audio_valid": torch.ones(n, dtype=torch.bool), "persons": { "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), "records": [{"id": "3"} for _ in range(n)], },