import torch
import torch.nn


class DummyModel(torch.nn.Module):
    def __init__(self):
        super().__init__()

    def foward(self, x):
        return 1