From a61ba5c52f368ac28329806c4e081dc51f5f03fb Mon Sep 17 00:00:00 2001 From: ncguilbeault Date: Fri, 20 Sep 2024 17:30:15 +0100 Subject: [PATCH] Scatter spikes in estimate place fields --- .../likelihoods/spiking_likelihood_glm.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/replay_trajectory_classification/likelihoods/spiking_likelihood_glm.py b/replay_trajectory_classification/likelihoods/spiking_likelihood_glm.py index ac185ac..ec676da 100644 --- a/replay_trajectory_classification/likelihoods/spiking_likelihood_glm.py +++ b/replay_trajectory_classification/likelihoods/spiking_likelihood_glm.py @@ -234,7 +234,8 @@ def estimate_place_fields( except ValueError: client = Client() design_matrix = client.scatter(np.asarray(design_matrix), broadcast=True) - results = [fit_glm(is_spike, design_matrix, penalty) for is_spike in spikes.T] + scattered_spikes = [client.scatter(is_spike) for is_spike in spikes.T] + results = [fit_glm(is_spike, design_matrix, penalty) for is_spike in scattered_spikes] results = dask.compute(*results) predict_matrix = make_spline_predict_matrix(design_info, place_bin_centers)