import logging

import torch
from datasets import Features, Sequence, Value, Array2D

from evaluate.module import EvaluationModule, EvaluationModuleInfo
import time

logging.basicConfig(level=logging.INFO)


class DummyMetric(EvaluationModule):
    def _info(self):
        return EvaluationModuleInfo(
            description="dummy metric for tests",
            citation="insert citation here",
            features=Features(
                {
                    "tensor": Sequence(Value("int64")),
                }
            ),
        )


metric = DummyMetric()
start_time = time.time()
metric.add_batch(
    tensor=torch.randint(0, 10, (1000, 7000)),
)
print(f"time={round(time.time()-start_time, 2)}s") # outputs time=32.41s