[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot]
2025-07-06 19:31:23 +00:00
committed by Adil Zouitine
parent b08149a113
commit 116059a43e
2 changed files with 8 additions and 8 deletions
+1 -1
View File
@@ -223,7 +223,7 @@ def test_complex_nested_observation():
reconstructed_batch = _default_transition_to_batch(transition)
# Check that all observation keys are preserved
original_obs_keys = {k for k in batch.keys() if k.startswith("observation.")}
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.")}
assert original_obs_keys == reconstructed_obs_keys