diff --git a/tests/processor/test_act_processor.py b/tests/processor/test_act_processor.py index b102aef80..b52df9a27 100644 --- a/tests/processor/test_act_processor.py +++ b/tests/processor/test_act_processor.py @@ -81,8 +81,8 @@ def test_make_act_processor_basic(): preprocessor, postprocessor = make_act_pre_post_processors(config, stats) # Check processor names - assert preprocessor.name == "robot_preprocessor" - assert postprocessor.name == "robot_postprocessor" + assert preprocessor.name == "policy_preprocessor" + assert postprocessor.name == "policy_postprocessor" # Check steps in preprocessor assert len(preprocessor.steps) == 4 diff --git a/tests/processor/test_diffusion_processor.py b/tests/processor/test_diffusion_processor.py index 7fcb92e3a..8ab477fab 100644 --- a/tests/processor/test_diffusion_processor.py +++ b/tests/processor/test_diffusion_processor.py @@ -84,8 +84,8 @@ def test_make_diffusion_processor_basic(): preprocessor, postprocessor = make_diffusion_pre_post_processors(config, stats) # Check processor names - assert preprocessor.name == "robot_preprocessor" - assert postprocessor.name == "robot_postprocessor" + assert preprocessor.name == "policy_preprocessor" + assert postprocessor.name == "policy_postprocessor" # Check steps in preprocessor assert len(preprocessor.steps) == 4 diff --git a/tests/processor/test_pi0_processor.py b/tests/processor/test_pi0_processor.py index 82db59921..2c8c0921a 100644 --- a/tests/processor/test_pi0_processor.py +++ b/tests/processor/test_pi0_processor.py @@ -110,8 +110,8 @@ def test_make_pi0_processor_basic(): ) # Check processor names - assert preprocessor.name == "robot_preprocessor" - assert postprocessor.name == "robot_postprocessor" + assert preprocessor.name == "policy_preprocessor" + assert postprocessor.name == "policy_postprocessor" # Check steps in preprocessor assert len(preprocessor.steps) == 6 diff --git a/tests/processor/test_sac_processor.py b/tests/processor/test_sac_processor.py index d60bb42a6..7a2d2d19c 100644 --- a/tests/processor/test_sac_processor.py +++ b/tests/processor/test_sac_processor.py @@ -86,8 +86,8 @@ def test_make_sac_processor_basic(): ) # Check processor names - assert preprocessor.name == "robot_preprocessor" - assert postprocessor.name == "robot_postprocessor" + assert preprocessor.name == "policy_preprocessor" + assert postprocessor.name == "policy_postprocessor" # Check steps in preprocessor assert len(preprocessor.steps) == 4 diff --git a/tests/processor/test_smolvla_processor.py b/tests/processor/test_smolvla_processor.py index 05bccec80..9a187fc7e 100644 --- a/tests/processor/test_smolvla_processor.py +++ b/tests/processor/test_smolvla_processor.py @@ -117,8 +117,8 @@ def test_make_smolvla_processor_basic(): ) # Check processor names - assert preprocessor.name == "robot_preprocessor" - assert postprocessor.name == "robot_postprocessor" + assert preprocessor.name == "policy_preprocessor" + assert postprocessor.name == "policy_postprocessor" # Check steps in preprocessor assert len(preprocessor.steps) == 6 diff --git a/tests/processor/test_tdmpc_processor.py b/tests/processor/test_tdmpc_processor.py index 14858b8b1..eb38143d7 100644 --- a/tests/processor/test_tdmpc_processor.py +++ b/tests/processor/test_tdmpc_processor.py @@ -89,8 +89,8 @@ def test_make_tdmpc_processor_basic(): ) # Check processor names - assert preprocessor.name == "robot_preprocessor" - assert postprocessor.name == "robot_postprocessor" + assert preprocessor.name == "policy_preprocessor" + assert postprocessor.name == "policy_postprocessor" # Check steps in preprocessor assert len(preprocessor.steps) == 4 diff --git a/tests/processor/test_vqbet_processor.py b/tests/processor/test_vqbet_processor.py index 891e88056..787a3d524 100644 --- a/tests/processor/test_vqbet_processor.py +++ b/tests/processor/test_vqbet_processor.py @@ -89,8 +89,8 @@ def test_make_vqbet_processor_basic(): ) # Check processor names - assert preprocessor.name == "robot_preprocessor" - assert postprocessor.name == "robot_postprocessor" + assert preprocessor.name == "policy_preprocessor" + assert postprocessor.name == "policy_postprocessor" # Check steps in preprocessor assert len(preprocessor.steps) == 4