From 753b996cda59bc22f9300dd733e099aed4f329a8 Mon Sep 17 00:00:00 2001 From: Steven Palma Date: Thu, 12 Feb 2026 21:25:39 +0100 Subject: [PATCH] test(rl): skip ci tests for resnet10 --- .../sac/reward_model/configuration_classifier.py | 2 +- tests/policies/hilserl/test_modeling_classifier.py | 13 +++++++++++++ tests/policies/test_sac_policy.py | 3 +++ 3 files changed, 17 insertions(+), 1 deletion(-) diff --git a/src/lerobot/policies/sac/reward_model/configuration_classifier.py b/src/lerobot/policies/sac/reward_model/configuration_classifier.py index 9b76b8037..879e3c1af 100644 --- a/src/lerobot/policies/sac/reward_model/configuration_classifier.py +++ b/src/lerobot/policies/sac/reward_model/configuration_classifier.py @@ -33,7 +33,7 @@ class RewardClassifierConfig(PreTrainedConfig): latent_dim: int = 256 image_embedding_pooling_dim: int = 8 dropout_rate: float = 0.1 - model_name: str = "helper2424/resnet10" + model_name: str = "helper2424/resnet10" # TODO: This needs to be updated. The model on the Hub doesn't call self.post_init() in its __init__, which is required by transformers v5 to set all_tied_weights_keys. The from_pretrained call fails when it tries to access this attribute during _finalize_model_loading. device: str = "cpu" model_type: str = "cnn" # "transformer" or "cnn" num_cameras: int = 2 diff --git a/tests/policies/hilserl/test_modeling_classifier.py b/tests/policies/hilserl/test_modeling_classifier.py index a572ea9e1..a62ef3ebb 100644 --- a/tests/policies/hilserl/test_modeling_classifier.py +++ b/tests/policies/hilserl/test_modeling_classifier.py @@ -14,6 +14,7 @@ # See the License for the specific language governing permissions and # limitations under the License. +import pytest import torch from lerobot.configs.types import FeatureType, NormalizationMode, PolicyFeature @@ -37,6 +38,9 @@ def test_classifier_output(): @require_package("transformers") +@pytest.mark.skip( + reason="helper2424/resnet10 needs to be updated to work with the latest version of transformers" +) def test_binary_classifier_with_default_params(): from lerobot.policies.sac.reward_model.modeling_classifier import Classifier @@ -78,6 +82,9 @@ def test_binary_classifier_with_default_params(): @require_package("transformers") +@pytest.mark.skip( + reason="helper2424/resnet10 needs to be updated to work with the latest version of transformers" +) def test_multiclass_classifier(): from lerobot.policies.sac.reward_model.modeling_classifier import Classifier @@ -117,6 +124,9 @@ def test_multiclass_classifier(): @require_package("transformers") +@pytest.mark.skip( + reason="helper2424/resnet10 needs to be updated to work with the latest version of transformers" +) def test_default_device(): from lerobot.policies.sac.reward_model.modeling_classifier import Classifier @@ -129,6 +139,9 @@ def test_default_device(): @require_package("transformers") +@pytest.mark.skip( + reason="helper2424/resnet10 needs to be updated to work with the latest version of transformers" +) def test_explicit_device_setup(): from lerobot.policies.sac.reward_model.modeling_classifier import Classifier diff --git a/tests/policies/test_sac_policy.py b/tests/policies/test_sac_policy.py index 6fad2979e..11499ce30 100644 --- a/tests/policies/test_sac_policy.py +++ b/tests/policies/test_sac_policy.py @@ -305,6 +305,9 @@ def test_sac_policy_with_visual_input(batch_size: int, state_dim: int, action_di [(1, 6, 6, "helper2424/resnet10"), (1, 6, 6, "facebook/convnext-base-224")], ) @pytest.mark.skipif(not TRANSFORMERS_AVAILABLE, reason="Transformers are not installed") +@pytest.mark.skip( + reason="helper2424/resnet10 needs to be updated to work with the latest version of transformers" +) def test_sac_policy_with_pretrained_encoder( batch_size: int, state_dim: int, action_dim: int, vision_encoder_name: str ):