a model which contains an error

model.py

import torch.nn as nn


class DemoErrorModel(nn.Module):
    def __init__(self, input_size=10, num_classes=2):
        super(DemoErrorModel, self).__init__()
        self.layer = nn.Linear(input_size, num_classes)
    
    def foward(self, X):
        return self.layer(X)