From a8a7b3c1b2706ebc1e21130abd5b89e56e3e24bd Mon Sep 17 00:00:00 2001 From: Matthew McDermott Date: Wed, 12 Jun 2024 23:04:14 -0400 Subject: [PATCH] Update function signature --- src/MEDS_tabular_automl/generate_summarized_reps.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/MEDS_tabular_automl/generate_summarized_reps.py b/src/MEDS_tabular_automl/generate_summarized_reps.py index 0745440..b0d66ac 100644 --- 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: