update setup

.gitignore

@@ -2,3 +2,6 @@
 *.json
 output*
 *checkpoint*
+build
+dist
+hapi.egg-info

hapi/__init__.py

@@ -12,4 +12,26 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from hapi.configure import Config as Config
+from hapi.configure import Config
+from hapi import callbacks
+from hapi import datasets
+from hapi import distributed
+from hapi import download
+from hapi import metrics
+from hapi import model
+from hapi import progressbar
+from hapi import text
+from hapi import vision
+
+__all__ = [
+    'Config',
+    'callbacks',
+    'datasets',
+    'distributed',
+    'download',
+    'metrics',
+    'model',
+    'progressbar',
+    'text',
+    'vision',
+]

hapi/vision/__init__.py

@@ -11,3 +11,8 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+from . import models
+from . import transforms
+
+__all__ = ["models", "transforms"]

setup.py

@@ -31,15 +31,26 @@ setuptools.setup(
     description="A Paddle High-level API that supports both static and dynamic execution modes (still under development)",
     url="https://github.com/PaddlePaddle/hapi",
     packages=[
-        'hapi', 'hapi.text', 'hapi.text.tokenizer', 'hapi.text.bert',
-        'hapi.text.bert.utils'
+        'hapi',
+        'hapi.datasets',
+        'hapi.text',
+        'hapi.text.tokenizer',
+        'hapi.text.bert',
+        'hapi.text.bert.utils',
+        'hapi.vision',
+        'hapi.vision.models',
+        'hapi.vision.transforms',
     ],
     package_dir={
         'hapi': './hapi',
+        'hapi.datasets': './hapi/datasets',
         'hapi.text': './hapi/text',
         'hapi.text.tokenizer': './hapi/text/tokenizer',
         'hapi.text.bert': './hapi/text/bert',
         'hapi.text.bert.utils': './hapi/text/bert/utils',
+        'hapi.vision': './hapi/vision',
+        'hapi.vision.models': './hapi/vision/models',
+        'hapi.vision.transforms': './hapi/vision/transforms',
     },
     platforms="any",
     license='Apache 2.0',