add auto option

bdpedigo · Apr 23, 2024 · f7e7fc6 · f7e7fc6
1 parent 3932f25
commit f7e7fc6
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/networkframe/networkframe.py b/networkframe/networkframe.py
@@ -1263,14 +1263,22 @@ def k_hop_aggregation(
         drop_non_numeric=True,
         n_jobs=-1,
         verbose=False,
-        engine="pandas",
+        engine="auto",
     ):
         if k < 0:
             raise ValueError("k must be non-negative.")
 
         if isinstance(aggregations, str):
             aggregations = [aggregations]
 
+        if engine == "auto":
+            if not all([isinstance(x, str) for x in aggregations]) or not all(
+                [x in ["mean", "sum", "std"] for x in aggregations]
+            ):
+                engine = "pandas"
+            else:
+                engine = "scipy"
+
         nodes = self.nodes
         if drop_non_numeric:
             nodes = nodes.select_dtypes(include=[np.number])