[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

light-curve/src/arrow_input.rs

@@ -44,7 +44,11 @@ pub(crate) struct ArrowLcsSchema {
     pub list_type: ArrowListType,
 }
 
-fn resolve_field(fields: &arrow_schema::Fields, field_ref: &ArrowFieldRef, role: &str) -> Res<usize> {
+fn resolve_field(
+    fields: &arrow_schema::Fields,
+    field_ref: &ArrowFieldRef,
+    role: &str,
+) -> Res<usize> {
     match field_ref {
         ArrowFieldRef::Index(i) => {
             if *i >= fields.len() {
@@ -56,15 +60,14 @@ fn resolve_field(fields: &arrow_schema::Fields, field_ref: &ArrowFieldRef, role:
                 Ok(*i)
             }
         }
-        ArrowFieldRef::Name(name) => fields
-            .iter()
-            .position(|f| f.name() == name)
-            .ok_or_else(|| {
+        ArrowFieldRef::Name(name) => {
+            fields.iter().position(|f| f.name() == name).ok_or_else(|| {
                 Exception::ValueError(format!(
                     "arrow_fields: field name {name:?} for {role} not found in struct fields {:?}",
                     fields.iter().map(|f| f.name()).collect::<Vec<_>>()
                 ))
-            }),
+            })
+        }
     }
 }
 

light-curve/src/features.rs

@@ -623,7 +623,11 @@ impl PyFeatureEvaluator {
                         "Null entries in the struct array are not supported".to_string(),
                     ));
                 }
-                for &col_idx in &[Some(t_idx), Some(m_idx), sigma_idx].into_iter().flatten().collect::<Vec<_>>() {
+                for &col_idx in &[Some(t_idx), Some(m_idx), sigma_idx]
+                    .into_iter()
+                    .flatten()
+                    .collect::<Vec<_>>()
+                {
                     if struct_arr.column(col_idx).null_count() > 0 {
                         return Err(Exception::NotImplementedError(
                             "Null values in data columns are not supported".to_string(),
@@ -677,9 +681,12 @@ impl PyFeatureEvaluator {
     }
 }
 
-
-fn parse_arrow_fields(arrow_fields: Option<&Bound<pyo3::types::PyList>>) -> Res<Option<Vec<ArrowFieldRef>>> {
-    let Some(list) = arrow_fields else { return Ok(None) };
+fn parse_arrow_fields(
+    arrow_fields: Option<&Bound<pyo3::types::PyList>>,
+) -> Res<Option<Vec<ArrowFieldRef>>> {
+    let Some(list) = arrow_fields else {
+        return Ok(None);
+    };
     let refs = list
         .iter()
         .map(|item| {
@@ -689,7 +696,8 @@ fn parse_arrow_fields(arrow_fields: Option<&Bound<pyo3::types::PyList>>) -> Res<
                 Ok(ArrowFieldRef::Name(name))
             } else {
                 Err(Exception::TypeError(
-                    "arrow_fields elements must be integers (indices) or strings (field names)".to_string(),
+                    "arrow_fields elements must be integers (indices) or strings (field names)"
+                        .to_string(),
                 ))
             }
         })
@@ -804,7 +812,15 @@ impl PyFeatureEvaluator {
         // Try Arrow path first
         if lcs.hasattr("__arrow_c_array__")? || lcs.hasattr("__arrow_c_stream__")? {
             let parsed = parse_arrow_fields(arrow_fields.as_ref())?;
-            return self.many_arrow(py, &lcs, fill_value, sorted, check, n_jobs, parsed.as_deref());
+            return self.many_arrow(
+                py,
+                &lcs,
+                fill_value,
+                sorted,
+                check,
+                n_jobs,
+                parsed.as_deref(),
+            );
         }
         // Fall back to list-of-tuples path
         let lcs: PyLcs<'py> = lcs.extract()?;

light-curve/tests/light_curve_ext/test_feature.py

@@ -855,29 +855,23 @@ def test_many_arrow_fields_nondefault_order():
 def test_many_arrow_fields_invalid_name():
     """arrow_fields with a nonexistent field name raises ValueError."""
     feature = lc.Amplitude()
-    arr = _make_arrow_lcs(
-        [gen_lc(10, rng=np.random.default_rng(5)) for _ in range(2)]
-    )
+    arr = _make_arrow_lcs([gen_lc(10, rng=np.random.default_rng(5)) for _ in range(2)])
     with pytest.raises(ValueError, match="not found"):
         feature.many(arr, sorted=True, arrow_fields=["t", "nonexistent"])
 
 
 def test_many_arrow_fields_index_out_of_range():
     """arrow_fields with an out-of-range index raises ValueError."""
     feature = lc.Amplitude()
-    arr = _make_arrow_lcs(
-        [gen_lc(10, rng=np.random.default_rng(6)) for _ in range(2)]
-    )
+    arr = _make_arrow_lcs([gen_lc(10, rng=np.random.default_rng(6)) for _ in range(2)])
     with pytest.raises(ValueError, match="out of range"):
         feature.many(arr, sorted=True, arrow_fields=[0, 99])
 
 
 def test_many_arrow_fields_wrong_count():
     """arrow_fields with 1 or 4 elements raises ValueError."""
     feature = lc.Amplitude()
-    arr = _make_arrow_lcs(
-        [gen_lc(10, rng=np.random.default_rng(7)) for _ in range(2)]
-    )
+    arr = _make_arrow_lcs([gen_lc(10, rng=np.random.default_rng(7)) for _ in range(2)])
     with pytest.raises(ValueError, match="2 .* or 3"):
         feature.many(arr, sorted=True, arrow_fields=["t"])
     with pytest.raises(ValueError, match="2 .* or 3"):
@@ -887,18 +881,14 @@ def test_many_arrow_fields_wrong_count():
 def test_many_arrow_fields_duplicate_fields():
     """arrow_fields pointing to the same field twice raises ValueError."""
     feature = lc.Amplitude()
-    arr = _make_arrow_lcs(
-        [gen_lc(10, rng=np.random.default_rng(8)) for _ in range(2)]
-    )
+    arr = _make_arrow_lcs([gen_lc(10, rng=np.random.default_rng(8)) for _ in range(2)])
     with pytest.raises(ValueError, match="different fields"):
         feature.many(arr, sorted=True, arrow_fields=["t", "t"])
 
 
 def test_many_arrow_fields_wrong_element_type():
     """Non-str, non-int element in arrow_fields raises TypeError."""
     feature = lc.Amplitude()
-    arr = _make_arrow_lcs(
-        [gen_lc(10, rng=np.random.default_rng(9)) for _ in range(2)]
-    )
+    arr = _make_arrow_lcs([gen_lc(10, rng=np.random.default_rng(9)) for _ in range(2)])
     with pytest.raises(TypeError, match="integers.*strings|strings.*integers"):
         feature.many(arr, sorted=True, arrow_fields=["t", 1.5])