Какую команду нужно использовать, чтобы задать размер изображения в игровом окне?

import pygame

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

pygame. init()

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

WIN_HEIGHT = 500

WIN_WIDTH = 500

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

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