Fixed a VisibleDeprecationWarning.

50a40ee1 · theos · 306e695a · 50a40ee1
Commit 50a40ee1 authored Jun 23, 2016 by theos
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

d2o/distributor_factory.py d2o/distributor_factory.py +3 -2

No files found.
--- a/d2o/distributor_factory.py
+++ b/d2o/distributor_factory.py
@@ -1712,8 +1712,9 @@ class _slicing_distributor(distributor):
                else:
                    new_data.data[:] = local_data.reshape(temp_local_shape)
            else:
-                if np.any(np.array(local_data.shape)[np.array(sliceified)] ==
-                          0):
+                if np.any(
+                    np.array(local_data.shape)[
+                        np.where(np.array(sliceified))] == 0):
                    temp = np.array(temp_local_shape)
                    temp[0] = 0
                    temp_local_shape = tuple(temp)