def METHOD_NAME(self):
    dataset = fo.Dataset()
    dataset.add_samples(
        [
            fo.Sample(filepath="image1.png"),
            fo.Sample(filepath="image2.png"),
            fo.Sample(filepath="image3.png"),
            fo.Sample(filepath="image4.png"),
            fo.Sample(filepath="image5.png"),
            fo.Sample(filepath="image6.png"),
            fo.Sample(filepath="image7.png"),
            fo.Sample(filepath="image8.png"),
            fo.Sample(filepath="image9.png"),
        ]
    )
    embeddings = np.random.randn(9, 4)
    fob.compute_similarity(
        dataset,
        embeddings=embeddings,
        brain_key="img_sim",
    )
    return dataset