fix format.

Signed-off-by: my-vegetable-has-exploded <[email protected]>

Author: my-vegetable-has-exploded

python/ray/data/_internal/plan.py

@@ -9,8 +9,9 @@
 from ray._private.internal_api import get_memory_info_reply, get_state_from_address
 from ray.data._internal.execution.interfaces import RefBundle
 from ray.data._internal.logical.interfaces import SourceOperator
-from ray.data._internal.logical.interfaces.operator import Operator
+from ray.data._internal.logical.interfaces.logical_operator import LogicalOperator
 from ray.data._internal.logical.interfaces.logical_plan import LogicalPlan
+from ray.data._internal.logical.interfaces.operator import Operator
 from ray.data._internal.logical.operators.read_operator import Read
 from ray.data._internal.stats import DatasetStats
 from ray.data._internal.util import unify_ref_bundles_schema
@@ -134,7 +135,7 @@ def generate_plan_string(
     ):
         """Traverse (DFS) the Plan DAG and
         return a string representation of the operators."""
-        if (including_source == False) and isinstance(op, SourceOperator):
+        if not including_source and isinstance(op, SourceOperator):
             return curr_str, depth
 
         curr_max_depth = depth

python/ray/data/dataset.py

@@ -5965,6 +5965,7 @@ def stats(self) -> str:
             return self._write_ds.stats()
         return self._get_stats_summary().to_string()
 
+    @PublicAPI(api_group=IM_API_GROUP)
     def explain(self) -> str:
         """Returns a string containing logical plan and physical plan.