I am facing `AttributeError: 'NoneType' object has no attribute 'shape'` while running example.py in develop branch. Please help! Python : 3.12.7 