Вы хотите читать .npz файл построчно, но что бы в результате получился такой же массив, как при чтении этого файла сразу за раз?
не совсем )
сейчас логика такая: читается файл датасета целиком (где записаны построчно предложения), затем они прогоняются через энкодер и конвертируются в numpy массив, а потом записыаются в npz
я хочу читать файл датасета построчно, прогонять каждую строку энкодером отдельно и конкантеновать их через numpy, а потом точно также записать результат в npz
трабл как раз в том (в колабе видно это), что в результате получаются разные npz файлы (содержимое test_2 отличается от reference)