From d2418c4fdf664b3605e25b0b1fb620b2653346aa Mon Sep 17 00:00:00 2001 From: Chaoming Wang Date: Tue, 27 Aug 2024 18:35:17 +0800 Subject: [PATCH] fix bugs --- dendritex/neurons/multi_compartment.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dendritex/neurons/multi_compartment.py b/dendritex/neurons/multi_compartment.py index e46b6bc..1c336a8 100644 --- a/dendritex/neurons/multi_compartment.py +++ b/dendritex/neurons/multi_compartment.py @@ -66,8 +66,8 @@ def diffusive_coupling(potentials, coo_ids, resistances): pre_ids = coo_ids[:, 0] post_ids = coo_ids[:, 1] diff = (potentials[..., pre_ids] - potentials[..., post_ids]) / resistances - outs[..., pre_ids] -= diff - outs[..., post_ids] += diff + outs = outs.at[..., pre_ids].add(-diff) + outs = outs.at[..., post_ids].add(diff) return outs