__init__.py 594 Bytes
Newer Older
1
2
from .alexnet import AlexNet_simplified
from .densenet import DenseNet
3
from .mine import MineModel
4

Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
5

6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# gets sphinx autodoc done right - don't remove it
def __appropriate__(*args):
    """Says object was actually declared here, an not on the import module.

    Parameters:

      *args: An iterable of objects to modify

    Resolves `Sphinx referencing issues
    <https://github.com/sphinx-doc/sphinx/issues/3048>`
    """

    for obj in args:
        obj.__module__ = __name__


Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
22
__appropriate__(AlexNet_simplified, DenseNet, MineModel)
23
__all__ = [_ for _ in dir() if not _.startswith("_")]