better versioning and import

setup.cfg

 [metadata]
 name = tensorpack
+version = attr: tensorpack.__version__
 author = TensorPack contributors
 author-email = ppwwyyxxc@gmail.com
-summary = Neural Network Toolbox on TensorFlow
-description-file = README.md
+description = Neural Network Toolbox on TensorFlow
+long_description = file: README.md
 url = https://github.com/ppwwyyxx/tensorpack
+keywords = tensorflow, deep learning, neural network
 license = Apache
 
-
-[files]
-packages =
-    tensorpack
+[options]
+zip_safe = False		    # dataset and __init__ use file
+packages = find:
 
 [wheel]
 universal = 1

setup.py

@@ -12,7 +12,6 @@ req = ['numpy',
 if sys.version_info.major == 2:
     req.extend(['subprocess32', 'functools32'])
 
-setup(version='0.1',
-      install_requires=req,
-      zip_safe=False    # dataset and __init__ use file
-      )
+# TODO:
+# setup_requires, extras_requires, scripts
+setup(install_requires=req)

tensorpack/__init__.py

@@ -2,8 +2,7 @@
 # File: __init__.py
 # Author: Yuxin Wu <ppwwyyxx@gmail.com>
 
-import numpy  # avoid https://github.com/tensorflow/tensorflow/issues/2034
-import cv2  # avoid https://github.com/tensorflow/tensorflow/issues/1924
+__version__ = '0.1'
 
 from tensorpack.train import *
 from tensorpack.models import *
@@ -12,6 +11,3 @@ from tensorpack.tfutils import *
 from tensorpack.callbacks import *
 from tensorpack.dataflow import *
 from tensorpack.predict import *
-
-if int(numpy.__version__.split('.')[1]) < 9:
-    logger.warn("Numpy < 1.9 could be extremely slow on some tasks.")

tensorpack/dataflow/tf_func.py

@@ -9,6 +9,8 @@ from .base import ProxyDataFlow
 
 """ This file was deprecated """
 
+__all__ = []
+
 
 class TFFuncMapper(ProxyDataFlow):
     def __init__(self, ds,

tensorpack/models/linearwrap.py

@@ -8,6 +8,8 @@ from types import ModuleType
 from ..utils import logger
 from .common import get_registered_layer
 
+__all__ = ['LinearWrap']
+
 
 class LinearWrap(object):
     """ A simple wrapper to easily create "linear" graph,

tensorpack/utils/utils.py

@@ -95,7 +95,7 @@ def get_dataset_path(*args):
             os.path.dirname(__file__), '..', 'dataflow', 'dataset'))
         if execute_only_once():
             from . import logger
-            logger.info("TENSORPACK_DATASET not set, using {} for dataset.".format(d))
+            logger.warn("TENSORPACK_DATASET not set, using {} for dataset.".format(d))
     assert os.path.isdir(d), d
     return os.path.join(d, *args)