Skip to content

Commit 3652a24

Browse files
authored
Merge pull request #1329 from zm711/numpy-updates
Update calls to NumPy dtypes being removed in NumPy 2.0
2 parents 340b022 + 2c0cca5 commit 3652a24

File tree

5 files changed

+7
-7
lines changed

5 files changed

+7
-7
lines changed

neo/core/analogsignal.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -718,7 +718,7 @@ def concatenate(self, *signals, overwrite=False, padding=False):
718718
if padding is not False:
719719
logger.warning('Signals will be padded using {}.'.format(padding))
720720
if padding is True:
721-
padding = np.NaN * units
721+
padding = np.nan * units
722722
if isinstance(padding, pq.Quantity):
723723
padding = padding.rescale(units).magnitude
724724
else:

neo/test/coretest/test_analogsignal.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1537,7 +1537,7 @@ def test_concatenate_padding_True(self):
15371537

15381538
result = signal1.concatenate(signal2, overwrite=False, padding=True)
15391539
assert_array_equal(
1540-
np.array([0, 1, 2, 3, np.NaN, np.NaN, np.NaN, 4, 5, 6]).reshape((-1, 1)),
1540+
np.array([0, 1, 2, 3, np.nan, np.nan, np.nan, 4, 5, 6]).reshape((-1, 1)),
15411541
result.magnitude)
15421542

15431543
def test_concatenate_padding_quantity(self):

neo/test/coretest/test_base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -853,7 +853,7 @@ def test_numpy_array_uint64(self):
853853

854854
def test_numpy_array_float(self):
855855
'''test to make sure float type numpy arrays are accepted'''
856-
value = np.array([1, 2, 3, 4, 5], dtype=np.float_)
856+
value = np.array([1, 2, 3, 4, 5], dtype=np.float64)
857857
self.base.annotate(data=value)
858858
result = {'data': value}
859859
self.assertDictEqual(result, self.base.annotations)
@@ -1041,7 +1041,7 @@ def test_numpy_scalar_uint64(self):
10411041

10421042
def test_numpy_scalar_float(self):
10431043
'''test to make sure float type numpy scalars are accepted'''
1044-
value = np.array(99, dtype=np.float_)
1044+
value = np.array(99, dtype=np.float64)
10451045
self.base.annotate(data=value)
10461046
result = {'data': value}
10471047
self.assertDictEqual(result, self.base.annotations)

neo/test/coretest/test_spiketrain.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -580,8 +580,8 @@ def test_defaults(self):
580580
train2 = _new_spiketrain(SpikeTrain, [3, 4, 5], units='sec', t_stop=10.0)
581581
assert_neo_object_is_compliant(train1)
582582
assert_neo_object_is_compliant(train2)
583-
self.assertEqual(train1.dtype, np.float_)
584-
self.assertEqual(train2.dtype, np.float_)
583+
self.assertEqual(train1.dtype, np.float64)
584+
self.assertEqual(train2.dtype, np.float64)
585585
self.assertEqual(train1.sampling_rate, 1.0 * pq.Hz)
586586
self.assertEqual(train2.sampling_rate, 1.0 * pq.Hz)
587587
self.assertEqual(train1.waveforms, None)

neo/test/iotest/test_nestio.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -356,7 +356,7 @@ def test_read_float(self):
356356
t_stop=500. * pq.ms,
357357
lazy=False, id_column_gdf=0, time_column_gdf=1)
358358
sts = seg.spiketrains
359-
self.assertTrue(all([s.magnitude.dtype == np.float_ for s in sts]))
359+
self.assertTrue(all([s.magnitude.dtype == np.float64 for s in sts]))
360360

361361
def test_values(self):
362362
"""

0 commit comments

Comments
 (0)