I want to know how to correct this error - TypeError: argument 2 to map() must support iteration at line 55 of my code? thank you

Question

#!/bin/python

import math
import os
import random
import re
import sys

#
# Complete the \'count_pebbles\' function below.
#
# The function is expected to return an INT