Python 3 Codigo de Hundir la flota

[Weng Weng]

Gente alguien tiene el código de Python 3 del juego hundir la flota?
Este me da fallo:

import numpy as np

from collections import Counter





def iguales(diccionario_1, diccionario_2):

    return diccionario_1.__eq__(diccionario_2)





def comprobar_barcos(oceano, cuenta_barcos):

    barcos_oceano = {}

    tamaño = 0

    for fila in range(oceano.shape[0]):

        for casilla in oceano[fila, :]:

            if casilla == 0:

                if tamaño > 1:

                    if tamaño in barcos_oceano:

                        barcos_oceano[tamaño] += 1

                    else:

                        barcos_oceano[tamaño] = 1

                tamaño = 0

            else:

                tamaño += 1

    for columna in range(oceano.shape[1]):

        for casilla in oceano[:, columna]:

            if casilla == 0:

                if tamaño > 1:

                    if tamaño in barcos_oceano:

                        barcos_oceano[tamaño] += 1

                    else:

                        barcos_oceano[tamaño] = 1

                tamaño = 0

            else:

                tamaño += 1



    return iguales(barcos_oceano, cuenta_barcos)





if __name__ == '__main__':

    respuestas = []

    numero_barcos = int(input())

    while numero_barcos > 0:

        lista_barcos = [int(x) for x in input().split()]

        cuenta_barcos = Counter(lista_barcos)

        tamaño_oceano = int(input())

        oceano = []

        for fila in range(tamaño_oceano):

            oceano.append([int(x) for x in input().split()])

        oceano = np.array(oceano)

        if comprobar_barcos(oceano, cuenta_barcos):

            respuestas.append('SI')

        else:

            respuestas.append('NO')

        numero_barcos = int(input())

    for respuesta in respuestas:

        print(respuesta)


Gracias foro