Ответ:

import pygame

#Здесь происходит инициализация

pygame.init()

# А здесь задаешь размеры

WIN_HEIGHT = 500

WIN_WIDTH = 500

pygame.display.set_mode((WIN_WIDTH, WIN_HEIGHT))

#Дальше твой код игры

Объяснение: