@rajul-iitkgp thanks for trying to fix this. I think the first is to remove the `print`-statement. I think in this case the code should stop, so when you catch the `AttributeError` - it would be great to reraise it with the error message you give (maybe then re-raise it as a `ValueError`). Thanks