chore(pylint): Reenable too-many-locals check (#16268)

Co-authored-by: John Bodley <john.bodley@airbnb.com>

.pylintrc

@@ -87,7 +87,6 @@ disable=
     import-outside-toplevel,
     raise-missing-from,
     too-few-public-methods,
-    too-many-locals,
     duplicate-code,
 
 [REPORTS]

superset/common/query_context.py

@@ -120,7 +120,7 @@ class QueryContext:
         df.reset_index(level=0, inplace=True)
         return df
 
-    def processing_time_offsets(
+    def processing_time_offsets(  # pylint: disable=too-many-locals
         self, df: pd.DataFrame, query_object: QueryObject,
     ) -> CachedTimeOffset:
         # ensure query_object is immutable

superset/dashboards/api.py

@@ -671,7 +671,7 @@ class DashboardRestApi(BaseSupersetModelRestApi):
     @event_logger.log_this_with_context(
         action=lambda self, *args, **kwargs: f"{self.__class__.__name__}.export",
         log_to_statsd=False,
-    )
+    )  # pylint: disable=too-many-locals
     def export(self, **kwargs: Any) -> Response:
         """Export dashboards
         ---

superset/dashboards/commands/importers/v1/utils.py

@@ -56,7 +56,7 @@ def build_uuid_to_id_map(position: Dict[str, Any]) -> Dict[str, int]:
     }
 
 
-def update_id_refs(
+def update_id_refs(  # pylint: disable=too-many-locals
     config: Dict[str, Any],
     chart_ids: Dict[str, int],
     dataset_info: Dict[str, Dict[str, Any]],

superset/datasets/api.py

@@ -415,7 +415,7 @@ class DatasetRestApi(BaseSupersetModelRestApi):
     @event_logger.log_this_with_context(
         action=lambda self, *args, **kwargs: f"{self.__class__.__name__}.export",
         log_to_statsd=False,
-    )
+    )  # pylint: disable=too-many-locals
     def export(self, **kwargs: Any) -> Response:
         """Export datasets
         ---

superset/examples/multiformat_time_series.py

@@ -32,7 +32,7 @@ from .helpers import (
 )
 
 
-def load_multiformat_time_series(
+def load_multiformat_time_series(  # pylint: disable=too-many-locals
     only_metadata: bool = False, force: bool = False
 ) -> None:
     """Loading time series data from a zip file in the repo"""

superset/utils/pandas_postprocessing.py

@@ -211,7 +211,7 @@ def _append_columns(
 
 
 @validate_column_args("index", "columns")
-def pivot(  # pylint: disable=too-many-arguments
+def pivot(  # pylint: disable=too-many-arguments,too-many-locals
     df: DataFrame,
     index: List[str],
     aggregates: Dict[str, Dict[str, Any]],

superset/utils/profiler.py

@@ -21,6 +21,7 @@ from unittest import mock
 from werkzeug.wrappers import Request, Response
 
 try:
+    # pylint: disable=import-error
     from pyinstrument import Profiler
 except ModuleNotFoundError:
     Profiler = None