another_module.py

import torch.nn as nn

def create_super_module(size):
    return nn.Linear(size, 1)