fix lints

modin-project · Jul 22, 2024 · bbf014e · bbf014e
1 parent ad02646
commit bbf014e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/modin/core/storage_formats/base/query_compiler.py b/modin/core/storage_formats/base/query_compiler.py
@@ -218,6 +218,15 @@ def default_to_pandas(self, pandas_op, *args, **kwargs) -> Self:
 
     @property
     def lazy_shape(self):
+        """
+        Whether either of the underlying dataframe's dimensions (row count/column count) are
+        computed lazily. If True, the frontend should avoid length/shape checks as much
+        as possible.
+
+        Returns
+        -------
+        bool
+        """
         return self.lazy_row_count or self.lazy_column_count
 
     _shape_hint = None

diff --git a/modin/core/storage_formats/pandas/query_compiler.py b/modin/core/storage_formats/pandas/query_compiler.py
@@ -1803,9 +1803,7 @@ def get_unique_level_values(index):
             new_index = (
                 get_unique_level_values(index)
                 if consider_index
-                else index
-                if isinstance(index, list)
-                else [index]
+                else index if isinstance(index, list) else [index]
             )
 
             new_columns = (