diff --git a/src/amuse/datamodel/memory_storage.py b/src/amuse/datamodel/memory_storage.py index 4312322c43..12b50002ea 100644 --- a/src/amuse/datamodel/memory_storage.py +++ b/src/amuse/datamodel/memory_storage.py @@ -632,7 +632,7 @@ def get_value(self, index): return self.quantity[index] def remove_indices(self, indices): - self.quantity._number = numpy.delete(self.quantity.number, indices) + self.quantity._number = numpy.delete(self.quantity.number, indices, axis=0) def has_units(self): return True @@ -679,7 +679,7 @@ def get_value(self, index): return self.values[index] def remove_indices(self, indices): - self.values = numpy.delete(self.values, indices) + self.values = numpy.delete(self.values, indices, axis=0) def has_units(self): return False @@ -749,7 +749,7 @@ def get_value(self, index): return value def remove_indices(self, indices): - self.values = numpy.delete(self.values, indices) + self.values = numpy.delete(self.values, indices, axis=0) def has_units(self): return False