Skip to content

Commit

Permalink
use vertex_slice property
Browse files Browse the repository at this point in the history
  • Loading branch information
Christian-B committed Oct 24, 2023
1 parent 3f57d09 commit 1ce0b76
Showing 1 changed file with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -494,8 +494,8 @@ def _fill_send_buffer_2d(self, key_base):
end_time_step = FecDataView.get_current_run_timesteps()
if first_time_step == end_time_step:
return
keys = get_field_based_keys(key_base, self._vertex_slice)
for atom in range(self._vertex_slice.n_atoms):
keys = get_field_based_keys(key_base, self.vertex_slice)
for atom in range(self.vertex_slice.n_atoms):
for tick in sorted(self._send_buffer_times[atom]):
if self._is_in_range(tick, first_time_step, end_time_step):
self._send_buffer.add_key(tick, keys[atom])
Expand All @@ -512,8 +512,8 @@ def _fill_send_buffer_1d(self, key_base):
end_time_step = FecDataView.get_current_run_timesteps()
if first_time_step == end_time_step:
return
keys = get_field_based_keys(key_base, self._vertex_slice)
key_list = [keys[atom] for atom in range(self._vertex_slice.n_atoms)]
keys = get_field_based_keys(key_base, self.vertex_slice)
key_list = [keys[atom] for atom in range(self.vertex_slice.n_atoms)]
for tick in sorted(self._send_buffer_times):
if self._is_in_range(tick, first_time_step, end_time_step):
self._send_buffer.add_keys(tick, key_list)
Expand Down

0 comments on commit 1ce0b76

Please sign in to comment.