extra requirements

Author: maxpumperla

.gitignore

@@ -77,4 +77,6 @@ java_param_sharing_model.h5
 temp.h5
 
 docs/sources
-docs/site
+docs/site
+
+.vscode

setup.py

@@ -9,7 +9,21 @@
       download_url='https://github.com/maxpumperla/elephas/tarball/0.4',
       author='Max Pumperla',
       author_email='[email protected]',
-      install_requires=['cython', 'keras', 'tensorflow', 'hyperas', 'flask', 'six', 'pyspark', 'pydl4j>=0.1.3'],
+      install_requires=['cython', 'tensorflow', 'keras', 'hyperas', 'flask', 'six', 'pyspark'],
+      extras_require={
+        'java': ['pydl4j>=0.1.3'],
+        'tests': ['pytest', 'pytest-pep8', 'pytest-cov', 'mock']
+    },
       packages=find_packages(),
       license='MIT',
-      zip_safe=False)
+      zip_safe=False,
+      classifiers=[
+        'Development Status :: 3 - Alpha',
+        'Intended Audience :: Developers',
+        'Environment :: Console',
+        'License :: OSI Approved :: Apache Software License',
+        'Operating System :: OS Independent',
+        'Programming Language :: Python',
+        'Programming Language :: Python :: 2',
+        'Programming Language :: Python :: 3'
+    ])