After looking at your notebook, there are a few things wrong there.
You are hard-coding the number of classes. You should not do that. It might pass you the tests, but it might fail down the line or in the autograder. As different unit tests might use different values.
You ignored two very important pieces of information given in the comments
# randomly sample sub-volume by sampling the corner voxel
# hint: make sure to leave enough room for the output dimensions!
You are neither using random sampling (the exact function to use has been mentioned to you in the dropdown “Hints” section of the Ex 1 introduction), nor are you leaving room for the output dimensions.