diff --git a/tests/processor/test_observation_processor.py b/tests/processor/test_observation_processor.py index 7b3d17109..f28e39245 100644 --- a/tests/processor/test_observation_processor.py +++ b/tests/processor/test_observation_processor.py @@ -260,35 +260,6 @@ def test_no_states_in_observation(): assert processed_obs == observation -def test_none_observation(): - """Test processor with None observation.""" - processor = StateProcessor() - - transition = (None, None, None, None, None, None, None) - result = processor(transition) - - assert result == transition - - -def test_serialization_methods(): - """Test serialization methods.""" - processor = StateProcessor() - - # Test get_config - config = processor.get_config() - assert isinstance(config, dict) - - # Test state_dict - state = processor.state_dict() - assert isinstance(state, dict) - - # Test load_state_dict (should not raise) - processor.load_state_dict(state) - - # Test reset (should not raise) - processor.reset() - - def test_complete_observation_processing(): """Test processing a complete observation with both images and states.""" processor = ObservationProcessor() diff --git a/tests/processor/test_pipeline.py b/tests/processor/test_pipeline.py index 936893847..644171c84 100644 --- a/tests/processor/test_pipeline.py +++ b/tests/processor/test_pipeline.py @@ -45,10 +45,7 @@ class MockStep: """Add a counter to the complementary_data.""" obs, action, reward, done, truncated, info, comp_data = transition - if comp_data is None: - comp_data = {} - else: - comp_data = dict(comp_data) # Make a copy + comp_data = {} if comp_data is None else dict(comp_data) # Make a copy comp_data[f"{self.name}_counter"] = self.counter self.counter += 1