input: 
  def create_linear_regression_model(input_dim: int, output_dim: int):
      """ Create a linear regression model class in Python which initialises the weights and bias,
      and computes the forward pass.
      >>> create_linear_regression_model(2, 1)
      <__main__.LinearRegressionModel object at 0x...>
      """
  
output:     class LinearRegressionModel:
          def __init__(self, input_dim, output_dim):
              self.weights = np.random.randn(input_dim, output_dim)
              self.bias = np.random.randn(output_dim)
          
          def forward(self, x):
              return np.dot(x, self.weights) + self.bias
      
      return LinearRegressionModel(input_dim, output_dim)