__init__.py 811 Bytes
Newer Older
1 2 3 4 5
#from DataShuffler import *
from .Layer import Layer
from .Conv2D import Conv2D
from .FullyConnected import FullyConnected
from .MaxPooling import MaxPooling
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
6
from .AveragePooling import AveragePooling
7
from .Dropout import Dropout
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
8
from .InputLayer import InputLayer
Tiago de Freitas Pereira's avatar
Scratch  
Tiago de Freitas Pereira committed
9

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

11
# gets sphinx autodoc done right - don't remove it
Tiago de Freitas Pereira's avatar
Tiago de Freitas Pereira committed
12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33
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__

__appropriate__(
    Layer,
    Conv2D,
    FullyConnected,
    MaxPooling,
    AveragePooling,
    Dropout,
    InputLayer
    )
34
__all__ = [_ for _ in dir() if not _.startswith('_')]
Tiago de Freitas Pereira's avatar
Scratch  
Tiago de Freitas Pereira committed
35