Feedback for BICEPS problem

Problem Link: BICEPS Problem - CodeChef

Feedback

It always shows runtime error for this problem. Armstrong Numbers code Biceps. I couldnt find a soultion even in the submitted ones kindly check the bug

@sukuthunder
can u send your code??

Sure, here is my code:
p = int(input())
for i in range(p):
N = int(input())
u =N
r = str(u)
t = len(r)
s = 0
if N == 0:
print(“Yes”)
while N > 0:
d = N % 10
k = d**t
s = s + k
N = N // 10
if s == u:
print(“yes”)
else:
print(“no”)

It works properly while running but when i submit my code it shows error