Skip to content
Snippets Groups Projects
Commit 178a09d3 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Build] always cast verification arrays to float32

parent 38d9ee72
No related branches found
No related tags found
No related merge requests found
......@@ -340,4 +340,7 @@ class DataflowBuildConfig:
+ self.verify_expected_output_npy
)
verify_expected_output_npy = np.load(self.verify_expected_output_npy)
return (verify_input_npy, verify_expected_output_npy)
return (
verify_input_npy.astype(np.float32),
verify_expected_output_npy.astype(np.float32),
)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment