geoguessr_bot/commands/embed_command.py

from dataclasses import dataclass

from geoguessr_bot.commands import AbstractCommand
from geoguessr_bot.retriever import AbstractImageEmbedder


@dataclass
class EmbedCommand(AbstractCommand):
    """Embed all images in a folder and save them in a .npy file
    """
    embedder: AbstractImageEmbedder
    images_folder: str
    output_path: str

    def run(self) -> None:
        self.embedder.embed_folder(self.images_folder, self.output_path)