80e30222e1
* Add py-bmtk and py-neurotools * py-bmtk: version bump * [@spackbot] updating style on behalf of heerener * Maybe the copyright needs to be extended to 2022 for the check to pass * Process review remarks * Update var/spack/repos/builtin/packages/py-neurotools/package.py Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com> Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
23 lines
773 B
Diff
23 lines
773 B
Diff
diff --git a/src/stgen.py b/src/stgen.py
|
|
index b818ea5..3cb45e6 100644
|
|
--- a/src/stgen.py
|
|
+++ b/src/stgen.py
|
|
@@ -227,7 +227,7 @@ class StGen:
|
|
number = min(5+numpy.ceil(2*n),100)
|
|
|
|
if number > 0:
|
|
- isi = self.rng.exponential(1.0/rate, number)*1000.0
|
|
+ isi = self.rng.exponential(1.0/rate, int(number)) * 1000.0
|
|
if number > 1:
|
|
spikes = numpy.add.accumulate(isi)
|
|
else:
|
|
@@ -301,7 +301,7 @@ class StGen:
|
|
number = min(5+numpy.ceil(2*n),100)
|
|
|
|
if number > 0:
|
|
- isi = self.rng.gamma(a, b, number)*1000.0
|
|
+ isi = self.rng.gamma(a, b, int(number))*1000.0
|
|
if number > 1:
|
|
spikes = numpy.add.accumulate(isi)
|
|
else:
|