Лично у меня были проблемы с размерностями массивов. Где-то в середине программы вместо строки может получиться матрица, но ошибки никакой не возникнет, так как в питоне предусмотрена работа методов с самыми разными типами входных данных.
Спасибо большое за помощь, все получилось. Протупил, не заметил, что макс числр итераций на последнем шаге нужно было выставить в 1е5, а у меня стояло 1е4, как по умолчанию