mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-23 04:30:10 +00:00
fix test
This commit is contained in:
+2
-2
@@ -20,8 +20,8 @@ def test_pi05_model_architecture():
|
||||
)
|
||||
|
||||
# Verify tokenizer max length is set correctly
|
||||
assert config.tokenizer_max_length == 48, (
|
||||
f"Expected tokenizer_max_length=48 for pi05, got {config.tokenizer_max_length}"
|
||||
assert config.tokenizer_max_length == 200, (
|
||||
f"Expected tokenizer_max_length=200 for pi05, got {config.tokenizer_max_length}"
|
||||
)
|
||||
print(f"✓ Tokenizer max length correctly set to {config.tokenizer_max_length}")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user