Fix ndarray imports in segment benches
use criterion::{criterion_group, criterion_main, Criterion};
-use ndarray::Array;
+use ndarray::{Array, Array2};
use rand::distributions::Standard;
use rand::Rng;
use tempdir::TempDir;
