Other files in the dataset (e.g., data_amp_10K.npy, xyz_coord.npy) load correctly, so the issue appears specific to ystress_all_new_10K.npy. It would be great if you could verify whether the dataset ...
There is a few test cases which are missing in the question,So write a solution by your own so that it should pass the test cases which are not mentioned there also,,i am writing one test case here ...