geoguessr_bot/retriever/random_embedder.py

from dataclasses import dataclass

import numpy as np
from PIL import Image

from geoguessr_bot.retriever import AbstractImageEmbedder


@dataclass
class RandomEmbedder(AbstractImageEmbedder):
    n_dim: int = 8

    def embed(self, image: Image) -> np.ndarray:
        """Embed an image
        """
        return np.random.rand(self.n_dim)