another_module.py

"""
Another module that is imported relatively.
This demonstrates the bug with relative imports in models with dots in their names.
"""
import torch


def custom_function(x):
    """A simple function that just returns the input unchanged."""
    return x


def another_helper():
    """Another helper function."""
    return "helper"