mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-20 11:09:59 +00:00
Add RTC initialization tests without config for PI0.5 and SmolVLA
Add test_pi05_rtc_initialization_without_rtc_config and test_smolvla_rtc_initialization_without_rtc_config to verify that policies can initialize without RTC config and that _rtc_enabled() returns False in this case. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
@@ -140,6 +140,25 @@ def test_pi05_rtc_initialization():
|
|||||||
print("✓ PI0.5 RTC initialization: Test passed")
|
print("✓ PI0.5 RTC initialization: Test passed")
|
||||||
|
|
||||||
|
|
||||||
|
@require_cuda
|
||||||
|
def test_pi05_rtc_initialization_without_rtc_config():
|
||||||
|
"""Test PI0.5 policy can initialize without RTC config."""
|
||||||
|
set_seed(42)
|
||||||
|
|
||||||
|
config = PI05Config(max_action_dim=7, max_state_dim=14, dtype="float32")
|
||||||
|
|
||||||
|
# Instantiate policy
|
||||||
|
policy = PI05Policy(config)
|
||||||
|
|
||||||
|
# Verify RTC processor is not initialized
|
||||||
|
assert hasattr(policy, "rtc_processor")
|
||||||
|
assert policy.rtc_processor is None
|
||||||
|
assert policy.model.rtc_processor is None
|
||||||
|
assert policy._rtc_enabled() is False
|
||||||
|
|
||||||
|
print("✓ PI0.5 RTC initialization without RTC config: Test passed")
|
||||||
|
|
||||||
|
|
||||||
@require_cuda
|
@require_cuda
|
||||||
def test_pi05_rtc_inference_with_prev_chunk():
|
def test_pi05_rtc_inference_with_prev_chunk():
|
||||||
"""Test PI0.5 policy inference with RTC and previous chunk."""
|
"""Test PI0.5 policy inference with RTC and previous chunk."""
|
||||||
|
|||||||
@@ -134,6 +134,25 @@ def test_smolvla_rtc_initialization():
|
|||||||
print("✓ SmolVLA RTC initialization: Test passed")
|
print("✓ SmolVLA RTC initialization: Test passed")
|
||||||
|
|
||||||
|
|
||||||
|
@require_cuda
|
||||||
|
def test_smolvla_rtc_initialization_without_rtc_config():
|
||||||
|
"""Test SmolVLA policy can initialize without RTC config."""
|
||||||
|
set_seed(42)
|
||||||
|
|
||||||
|
config = SmolVLAConfig(max_action_dim=7, chunk_size=50)
|
||||||
|
|
||||||
|
# Instantiate policy
|
||||||
|
policy = SmolVLAPolicy(config)
|
||||||
|
|
||||||
|
# Verify RTC processor is not initialized
|
||||||
|
assert hasattr(policy, "rtc_processor")
|
||||||
|
assert policy.rtc_processor is None
|
||||||
|
assert policy.model.rtc_processor is None
|
||||||
|
assert policy._rtc_enabled() is False
|
||||||
|
|
||||||
|
print("✓ SmolVLA RTC initialization without RTC config: Test passed")
|
||||||
|
|
||||||
|
|
||||||
@require_cuda
|
@require_cuda
|
||||||
@pytest.mark.skipif(True, reason="Requires pretrained SmolVLA model weights")
|
@pytest.mark.skipif(True, reason="Requires pretrained SmolVLA model weights")
|
||||||
def test_smolvla_rtc_inference_with_prev_chunk():
|
def test_smolvla_rtc_inference_with_prev_chunk():
|
||||||
|
|||||||
Reference in New Issue
Block a user