update fast_kde_2d benchmark and changelog

.azure-pipelines/azure-pipelines-external.yml

@@ -119,7 +119,7 @@ jobs:
     displayName: 'pydocstyle'
 
   - script: |
-      python -m black --check arviz examples
+      python -m black --check arviz examples asv_benchmarks
     displayName: 'black'
 
   - task: PublishTestResults@2

CHANGELOG.md

@@ -16,6 +16,8 @@
 * Fixed `TypeError` in `transform` argument of `plot_density` and `plot_forest` when `InferenceData` is a list or tuple (#1121)
 * Fixed overlaid pairplots issue (#1135)
 * Update Docker building steps (#1127)
+* Updated benchmarks and moved to asv_benchmarks/benchmarks (#1142)
+* Moved `_fast_kde`, `_fast_kde_2d`, `get_bins` and `_sturges_formula` to `numeric_utils` and `get_coords` to `utils` (#1142)
 ### Deprecation
 
 ### Documentation

CONTRIBUTING.md

@@ -192,7 +192,7 @@ tools:
 
   ```bash
   $ pip install black
-  $ black arviz/ examples/
+  $ black arviz/ examples/ asv_benchmarks/
   ```
 
 * Your code passes pylint

asv_benchmarks/benchmarks/benchmarks.py

@@ -34,7 +34,7 @@ def setup(self, numba_flag):
         else:
             az.Numba.disable_numba()
 
-    def time_variance(self, numba_flag):
+    def time_variance_2d(self, numba_flag):
         stats_variance_2d(self.data)
 
 
@@ -54,11 +54,11 @@ def time_circ_std(self, numba_flag):
 
 
 class Fast_Kde_1d:
-    params = [(True, False), (10**5, 10**6, 10**7)]
+    params = [(True, False), (10 ** 5, 10 ** 6, 10 ** 7)]
     param_names = ("Numba", "n")
 
     def setup(self, numba_flag, n):
-        self.x = np.random.randn(n//10, 10)
+        self.x = np.random.randn(n // 10, 10)
         if numba_flag:
             az.Numba.enable_numba()
         else:
@@ -67,21 +67,23 @@ def setup(self, numba_flag, n):
     def time_fast_kde_normal(self, numba_flag, n):
         _fast_kde(self.x)
 
+
 class Fast_KDE_2d:
-    params = [(True, False), (10**5, 10**6)]
-    param_names = ("Numba", "n")
+    params = [(True, False), ((100, 10 ** 4), (10 ** 4, 100), (1000, 1000))]
+    param_names = ("Numba", "shape")
 
-    def setup(self, numba_flag, n):
-        self.x = np.random.randn(n//10, 10)
-        self.y = np.random.randn(n//10, 10)
+    def setup(self, numba_flag, shape):
+        self.x = np.random.randn(*shape)
+        self.y = np.random.randn(*shape)
         if numba_flag:
             az.Numba.enable_numba()
         else:
             az.Numba.disable_numba()
 
-    def time_fast_kde_2d(self, numba_flag, n):
+    def time_fast_kde_2d(self, numba_flag, shape):
         _fast_kde_2d(self.x, self.y)
 
+
 class Atleast_Nd:
     params = ("az.utils", "numpy")
     param_names = ("source",)

scripts/lint.sh

@@ -9,7 +9,11 @@ python -m pydocstyle --convention=numpy ${SRC_DIR}/arviz/
 echo "Success!"
 
 echo "Checking code style with black..."
+<<<<<<< HEAD
 python -m black -l 100 --check ${SRC_DIR}/arviz/ ${SRC_DIR}/examples/
+=======
+python -m black -l 100 --check ${SRC_DIR}/arviz/ ${SRC_DIR}/examples/ ${SRC_DIR}/asv_benchmarks/
+>>>>>>> update fast_kde_2d benchmark and changelog
 echo "Success!"
 
 echo "Checking code style with pylint..."