I'm trying to recreate your test steps mentioned in test_run.ipynb. I found the example file you used for testing is not provided in repository. I tried to use another csv file and check but, I'm getting "KeyError: '0'" error message with no head start to troubleshoot.
Please provide the test file used in the testing exercise.
Thanks.