index XXX is out of bounds for axis 1 with size 10

[jli113]

Traceback (most recent call last):
File "/home/m/prettyNeatWann/wann_train.py", line 280, in
main(args)
File "/home/m/prettyNeatWann/wann_train.py", line 256, in main
slave()
File "/home/m/prettyNeatWann/wann_train.py", line 196, in slave
result = task.getFitness(wVec,aVec,hyp,seed=seed) # process it
File "/home/m/prettyNeatWann/domain/wann_task_gym.py", line 90, in getFitness
reward[iRep,iVal] = self.testInd(wMat, aVec, seed=seed+iRep,view=view)
File "/home/m/prettyNeatWann/domain/task_gym.py", line 91, in testInd
state, reward, done, info = self.env.step(action)
File "/home/m/prettyNeatWann/domain/classify_gym.py", line 67, in step
log_likelihood = -np.log(action[range(m),y])
IndexError: index 780017 is out of bounds for axis 1 with size 10