# Lint as: python3
"""This script is used to run the amaranth.ml module."""

from amaranth.ml import train


def main():
  train.main()


if __name__ == '__main__':
  main()

# Lint as: python3
"""This script is used to run the amaranth.ml module as an executable.

For now, this script just delegates it's main to amaranth.ml.train
"""

from amaranth.ml import train


def main():
  train.main()


if __name__ == '__main__':
  main()

Update docstring for amaranth.ml.__main__