__init__.py

"""
This is here so that the RETURNN repo root directory can be imported as a submodule.
This was used for the old flat code file structure, for usage like::

    import returnn.TFUtil

We want to support the same code.
"""


from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import os
import sys

if globals().get("__package__", None) is None:
  __package__ = __name__  # https://www.python.org/dev/peps/pep-0366/ # pylint: disable=redefined-builtin

sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

if __name__ == "returnn":
  # This is a bit tricky. `import returnn` in here will not work, as this is the reference to ourselves!
  # Thus delete this ref, to enforce a reload of the right module.
  # For the user, this will actually replace the `returnn` module by the right one.
  # But this should be compatible.
  sys.modules.pop("returnn", None)
  # It is important that we have some `import returnn` below.

from returnn.__old_mod_loader__ import setup as _old_module_loader_setup  # nopep8 # pylint: disable=wrong-import-position

# noinspection PyUnboundLocalVariable
_old_module_loader_setup(package_name=__package__, modules=globals())