Fix Notebook Controller bug

clement-moulin-frier · Mar 18, 2024 · abc8704 · abc8704
1 parent 1f592d8
commit abc8704
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vivarium/simulator/grpc_server/simulator_server.py b/vivarium/simulator/grpc_server/simulator_server.py
@@ -80,9 +80,10 @@ def SetState(self, request, context):
                                                    proto_to_ndarray(request.value))
         return Empty()
 
+    # TODO : Clean the function in the future to prevent having self everywhere 
     def Step(self, request, context):
         assert not self.simulator.is_started()
-        self.simulator.run(threaded=False, num_loops=1)
+        self.simulator.state, self.simulator.neighbors = self.simulator.step(self.simulator.state, self.simulator.neighbors)
         return state_to_proto(self.simulator.state)