Add minimal package versions for requirements.txt (#804)

* Minimal requirements.txt update

* small pylint fix

* update optional pip

arviz/plots/loopitplot.py

@@ -149,8 +149,8 @@ def plot_loo_pit(
     if plot_unif_kwargs is None:
         plot_unif_kwargs = {}
     light_color = rgb_to_hsv(to_rgb(plot_kwargs.get("color")))
-    light_color[1] /= 2
-    light_color[2] += (1 - light_color[2]) / 2
+    light_color[1] /= 2  # pylint: disable=unsupported-assignment-operation
+    light_color[2] += (1 - light_color[2]) / 2  # pylint: disable=unsupported-assignment-operation
     plot_unif_kwargs.setdefault("color", hsv_to_rgb(light_color))
     plot_unif_kwargs.setdefault("alpha", 0.5)
     plot_unif_kwargs.setdefault("linewidth", 0.6 * linewidth)

requirements-optional.txt

@@ -0,0 +1 @@
+numba

requirements.txt

@@ -1,6 +1,6 @@
 matplotlib>=3.0
-numpy
-scipy
-pandas
-xarray
+numpy>=1.12
+scipy>=0.19
+pandas>=0.23
+xarray>=0.11
 netcdf4

setup.py

@@ -9,38 +9,53 @@
 
 
 PROJECT_ROOT = os.path.dirname(os.path.realpath(__file__))
-REQUIREMENTS_FILE = os.path.join(PROJECT_ROOT, 'requirements.txt')
-README_FILE = os.path.join(PROJECT_ROOT, 'README.md')
-VERSION_FILE = os.path.join(PROJECT_ROOT, 'arviz', '__init__.py')
+REQUIREMENTS_FILE = os.path.join(PROJECT_ROOT, "requirements.txt")
+REQUIREMENTS_OPTIONAL_FILE = os.path.join(PROJECT_ROOT, "requirements-optional.txt")
+REQUIREMENTS_DEV_FILE = os.path.join(PROJECT_ROOT, "requirements-dev.txt")
+README_FILE = os.path.join(PROJECT_ROOT, "README.md")
+VERSION_FILE = os.path.join(PROJECT_ROOT, "arviz", "__init__.py")
 
 
 def get_requirements():
     with codecs.open(REQUIREMENTS_FILE) as buff:
         return buff.read().splitlines()
 
 
+def get_requirements_dev():
+    with codecs.open(REQUIREMENTS_DEV_FILE) as buff:
+        return buff.read().splitlines()
+
+
+def get_requirements_optional():
+    with codecs.open(REQUIREMENTS_OPTIONAL_FILE) as buff:
+        return buff.read().splitlines()
+
+
 def get_long_description():
-    with codecs.open(README_FILE, 'rt') as buff:
+    with codecs.open(README_FILE, "rt") as buff:
         return buff.read()
 
+
 def get_version():
-    lines = open(VERSION_FILE, 'rt').readlines()
+    lines = open(VERSION_FILE, "rt").readlines()
     version_regex = r"^__version__ = ['\"]([^'\"]*)['\"]"
     for line in lines:
         mo = re.search(version_regex, line, re.M)
         if mo:
             return mo.group(1)
-    raise RuntimeError('Unable to find version in %s.' % (VERSION_FILE,))
+    raise RuntimeError("Unable to find version in %s." % (VERSION_FILE,))
+
 
 setup(
-    name='arviz',
+    name="arviz",
     version=get_version(),
-    description='Exploratory analysis of Bayesian models',
-    author='ArviZ Developers',
+    description="Exploratory analysis of Bayesian models",
+    author="ArviZ Developers",
     url="http://github.com/arviz-devs/arviz",
     packages=find_packages(),
     install_requires=get_requirements(),
+    extras_require=dict(test=get_requirements_dev(), all=get_requirements_optional()),
     long_description=get_long_description(),
-    long_description_content_type='text/markdown',
+    long_description_content_type="text/markdown",
     include_package_data=True,
 )