fix(test): linting issue

This commit is contained in:
Adil Zouitine
2025-07-06 21:34:14 +02:00
parent 116059a43e
commit 730c7b2f35
+1 -1
View File
@@ -224,7 +224,7 @@ def test_complex_nested_observation():
# Check that all observation keys are preserved
original_obs_keys = {k for k in batch if k.startswith("observation.")}
reconstructed_obs_keys = {k for k in reconstructed_batch.keys() if k.startswith("observation.")}
reconstructed_obs_keys = {k for k in reconstructed_batch if k.startswith("observation.")}
assert original_obs_keys == reconstructed_obs_keys