fix unit test build
fix unit test build

adamcrume/tensorflow-rust,adamcrume/tensorflow-rust,tensorflow/rust,tensorflow/rust,adamcrume/tensorflow-rust,tensorflow/rust,adamcrume/tensorflow-rust,tensorflow/rust