lint.

dmlc · Jan 18, 2021 · 142de5d · 142de5d
1 parent 76bff73
commit 142de5d
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/python-package/xgboost/dask.py b/python-package/xgboost/dask.py
@@ -1029,10 +1029,11 @@ def dispatched_predict(
                     else:
                         groups = predt.shape[1]
                         columns = predt.shape[2]
+                    # pylint: disable=no-member
                     ret = (
                         [dask.delayed(predt), groups, columns],
                         order,
-                    )  # pylint: disable=no-member
+                    )
                 elif pred_interactions and predt.size != local_part.num_row():
                     assert len(predt.shape) in (3, 4)
                     if len(predt.shape) == 3:
@@ -1041,17 +1042,19 @@ def dispatched_predict(
                     else:
                         groups = predt.shape[1]
                         columns = predt.shape[2]
+                    # pylint: disable=no-member
                     ret = (
                         [dask.delayed(predt), groups, columns],
                         order,
-                    )  # pylint: disable=no-member
+                    )
                 else:
                     assert len(predt.shape) == 1 or len(predt.shape) == 2
                     columns = 1 if len(predt.shape) == 1 else predt.shape[1]
+                    # pylint: disable=no-member
                     ret = (
                         [dask.delayed(predt), columns],
                         order,
-                    )  # pylint: disable=no-member
+                    )
                 predictions.append(ret)
 
             return predictions