I encountered a ValueError: axes don't match array while running the example_variant.py. Actual Behavior: The script fails with the following error: raceback (most recent call last): File ...