__init__.py 560 Bytes
Newer Older
Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
1
2
3
from .center_loss import CenterLoss
from .center_loss import CenterLossLayer

Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
4

5
# gets sphinx autodoc done right - don't remove it
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
6
def __appropriate__(*args):
7
    """Says object was actually declared here, an not on the import module.
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
8

Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
9
    Parameters:
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
10

Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
11
      *args: An iterable of objects to modify
Tiago de Freitas Pereira's avatar
Scratch  
Tiago de Freitas Pereira committed
12

Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
13
14
15
    Resolves `Sphinx referencing issues
    <https://github.com/sphinx-doc/sphinx/issues/3048>`
    """
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
16

17
18
    for obj in args:
        obj.__module__ = __name__
Tiago de Freitas Pereira's avatar
Scratch  
Tiago de Freitas Pereira committed
19
20


Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
21
__appropriate__(CenterLoss, CenterLossLayer)
22
__all__ = [_ for _ in dir() if not _.startswith("_")]