simplify reset_graph_elements

Author: Christian-B

spynnaker/pyNN/models/neuron/population_vertex.py

@@ -42,8 +42,6 @@
 from pacman.model.resources import AbstractSDRAM, MultiRegionSDRAM
 from pacman.utilities.utility_calls import get_n_bits
 
-from spinn_front_end_common.abstract_models import (
-    AbstractCanReset)
 from spinn_front_end_common.interface.buffer_management\
     .recording_utilities import (
        get_recording_header_size, get_recording_data_constant_size)
@@ -173,7 +171,7 @@ def _is_structural(dynamics: AbstractSynapseDynamics
 
 class PopulationVertex(
         PopulationApplicationVertex, AbstractAcceptsIncomingSynapses,
-        AbstractCanReset, SupportsStructure):
+        SupportsStructure):
     """
     Underlying vertex model for Neural Populations.
     """
@@ -1348,7 +1346,6 @@ def __str__(self) -> str:
     def __repr__(self) -> str:
         return self.__str__()
 
-    @overrides(AbstractCanReset.reset_to_first_timestep)
     def reset_to_first_timestep(self) -> None:
         # Reset state variables
         self.__state_variables.copy_into(self.__initial_state_variables)

spynnaker/pyNN/spinnaker.py

@@ -42,7 +42,8 @@
 from spynnaker.pyNN import model_binaries
 from spynnaker.pyNN.config_setup import add_spynnaker_cfg, SPYNNAKER_CFG
 from spynnaker.pyNN.models.recorder import Recorder
-from spynnaker.pyNN.models.neuron import AbstractPyNNNeuronModel
+from spynnaker.pyNN.models.neuron import (
+    AbstractPyNNNeuronModel, PopulationVertex)
 from spynnaker.pyNN.data import SpynnakerDataView
 from spynnaker.pyNN.data.spynnaker_data_writer import SpynnakerDataWriter
 from spynnaker.pyNN.extra_algorithms import (
@@ -433,3 +434,9 @@ def _execute_buffer_extractor(self) -> None:
         if not get_config_bool("Machine", "virtual_board"):
             with NeoBufferDatabase() as db:
                 db.write_t_stop()
+
+    @overrides(AbstractSpinnakerBase._execute_buffer_extractor)
+    def _reset_graph_elements(self) -> None:
+         for vertex in self._data_writer.get_vertices_by_type(
+                 PopulationVertex):
+            vertex.reset_to_first_timestep()