Hi,
I don’t think it is None here.
What is the full stack trace you see?
Also can you split all the computations in the line that fails to different lines so that the stack trace can point to the explicit op that failed.