in numpy/src/main/scala/com/spotify/featran/numpy/NumPy.scala [173:190]
def write[@specialized(Int, Long, Float, Double) T: NumPyType](
out: OutputStream,
data: Array[T],
shape: Seq[Int] = Seq.empty
): Unit = {
val dims = if (shape.isEmpty) {
Seq(data.length)
} else {
require(
data.length == shape.product,
s"Invalid shape, ${shape.mkString(" * ")} != ${data.length}"
)
shape
}
writeHeader(out, dims)
writeData(out, data)
out.flush()
}