fix Ci issues

Author: ShreyeshArangath

crates/core/src/metrics.rs

@@ -89,9 +89,7 @@ impl PyMetric {
         match ts.value() {
             Some(dt) => {
                 let nanos = dt.timestamp_nanos_opt().ok_or_else(|| {
-                    PyErr::new::<pyo3::exceptions::PyOverflowError, _>(
-                        "timestamp out of range",
-                    )
+                    PyErr::new::<pyo3::exceptions::PyOverflowError, _>("timestamp out of range")
                 })?;
                 let datetime_mod = py.import("datetime")?;
                 let datetime_cls = datetime_mod.getattr("datetime")?;
@@ -126,8 +124,12 @@ impl PyMetric {
             MetricValue::SpilledBytes(c) => Ok(Some(c.value().into_pyobject(py)?.into_any())),
             MetricValue::SpilledRows(c) => Ok(Some(c.value().into_pyobject(py)?.into_any())),
             MetricValue::CurrentMemoryUsage(g) => Ok(Some(g.value().into_pyobject(py)?.into_any())),
-            MetricValue::Count { count, .. } => Ok(Some(count.value().into_pyobject(py)?.into_any())),
-            MetricValue::Gauge { gauge, .. } => Ok(Some(gauge.value().into_pyobject(py)?.into_any())),
+            MetricValue::Count { count, .. } => {
+                Ok(Some(count.value().into_pyobject(py)?.into_any()))
+            }
+            MetricValue::Gauge { gauge, .. } => {
+                Ok(Some(gauge.value().into_pyobject(py)?.into_any()))
+            }
             MetricValue::Time { time, .. } => Ok(Some(time.value().into_pyobject(py)?.into_any())),
             MetricValue::StartTimestamp(ts) | MetricValue::EndTimestamp(ts) => {
                 Self::timestamp_to_pyobject(py, ts)

python/datafusion/plan.py

@@ -19,13 +19,13 @@
 
 from __future__ import annotations
 
-import datetime
-
 from typing import TYPE_CHECKING, Any
 
 import datafusion._internal as df_internal
 
 if TYPE_CHECKING:
+    import datetime
+
     from datafusion.context import SessionContext
 
 __all__ = [

python/tests/test_plans.py

@@ -18,7 +18,6 @@
 import datetime
 
 import pytest
-
 from datafusion import (
     ExecutionPlan,
     LogicalPlan,
@@ -67,7 +66,8 @@ def test_metrics_tree_walk() -> None:
         if ms.output_rows is not None:
             output_rows_by_op[name] = ms.output_rows
 
-    # The filter passes rows where column1 > 1, so exactly 2 rows from (1,'a'),(2,'b'),(3,'c').
+    # The filter passes rows where column1 > 1, so exactly
+    # 2 rows from (1,'a'),(2,'b'),(3,'c').
     # At least one operator must report exactly 2 output rows (the filter).
     assert 2 in output_rows_by_op.values(), (
         f"Expected an operator with output_rows=2, got {output_rows_by_op}"