Skip to content

Commit a9bdcbe

Browse files
authored
FEAT-#4527: Add Modin logging to AxisPartition and BlockPartition classes (#7079)
Signed-off-by: arunjose696 <[email protected]>
1 parent 93b4e2a commit a9bdcbe

File tree

3 files changed

+7
-4
lines changed

3 files changed

+7
-4
lines changed

modin/core/dataframe/base/partitioning/axis_partition.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,10 @@
1616
from abc import ABC, abstractmethod
1717
from typing import Any, Callable, Iterable, Optional, Tuple, Union
1818

19+
from modin.logging import ClassLogger
1920

20-
class BaseDataframeAxisPartition(ABC): # pragma: no cover
21+
22+
class BaseDataframeAxisPartition(ABC, ClassLogger): # pragma: no cover
2123
"""
2224
An abstract class that represents the parent class for any axis partition class.
2325

modin/core/dataframe/pandas/partitioning/partition.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,11 +23,11 @@
2323
from pandas.util import cache_readonly
2424

2525
from modin.core.storage_formats.pandas.utils import length_fn_pandas, width_fn_pandas
26-
from modin.logging import get_logger
26+
from modin.logging import ClassLogger, get_logger
2727
from modin.pandas.indexing import compute_sliced_len
2828

2929

30-
class PandasDataframePartition(ABC): # pragma: no cover
30+
class PandasDataframePartition(ABC, ClassLogger): # pragma: no cover
3131
"""
3232
An abstract class that is base for any partition class of ``pandas`` storage format.
3333

modin/core/execution/ray/implementations/pandas_on_ray/partitioning/partition.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@
3030
MetaListHook,
3131
)
3232
from modin.core.execution.ray.common.utils import ObjectIDType
33-
from modin.logging import get_logger
33+
from modin.logging import disable_logging, get_logger
3434
from modin.pandas.indexing import compute_sliced_len
3535
from modin.utils import _inherit_docstrings
3636

@@ -93,6 +93,7 @@ def __init__(
9393
)
9494
)
9595

96+
@disable_logging
9697
def __del__(self):
9798
"""Unsubscribe from DeferredExecution."""
9899
if isinstance(self._data_ref, DeferredExecution):

0 commit comments

Comments
 (0)