Update function signature

mmcdermott · Jun 13, 2024 · a8a7b3c · a8a7b3c
1 parent afdc9b1
commit a8a7b3c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/MEDS_tabular_automl/generate_summarized_reps.py b/src/MEDS_tabular_automl/generate_summarized_reps.py
@@ -16,7 +16,7 @@
 )
 
 
-def sparse_aggregate(sparse_matrix: sparray, agg: str) -> csr_array:
+def sparse_aggregate(sparse_matrix: sparray, agg: str) -> np.ndarray | coo_array:
     """Aggregates values in a sparse matrix according to the specified method.
 
     Args: