mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-23 12:40:08 +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
|
# Verify tokenizer max length is set correctly
|
||||||
assert config.tokenizer_max_length == 48, (
|
assert config.tokenizer_max_length == 200, (
|
||||||
f"Expected tokenizer_max_length=48 for pi05, got {config.tokenizer_max_length}"
|
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}")
|
print(f"✓ Tokenizer max length correctly set to {config.tokenizer_max_length}")
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user