Help me in solving LPYAS151 problem

help
,
1

My issue

the answer is correct still not getting submitted

My code

N = int(input("Enter a number: "))
num_str = str(N)
num_digits  = len(num_str)
sum_of_powers = sum(int(digit) ** num_digits for digit in num_str)
if sum_of_powers == N:
    print("Armstrong")
    
else:
    print("Not Armstrong")

Learning course: Algorithmic Problem Solving
Problem Link: https://www.codechef.com/learn/course/mitwpu-problem-solving/MWPURM201/problems/LPYAS151