Niko/ActiveBOROS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

publisher doesnt work2

Browse Source
This commit is contained in:
Niko Feith 2023-03-22 13:15:35 +01:00
parent f25d1ff559
commit 0de3eaa360
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/active_bo_ros/active_bo_ros/active_rl_service.py
View File

@ -10,6 +10,7 @@ from rclpy.callback_groups import ReentrantCallbackGroup
from active_bo_ros.ReinforcementLearning.ContinuousMountainCar import Continuous_MountainCarEnv
import numpy as np
import time
class ActiveRLService(Node):
@ -33,13 +34,18 @@ class ActiveRLService(Node):
callback_group=sub_callback_group)
self.eval_response_received = False
self.eval_response = None
self.eval_response_received_first = False
self.env = Continuous_MountainCarEnv(render_mode='rgb_array')
self.distance_penalty = 0
def active_rl_eval_callback(self, response):
self.eval_response = response
self.eval_response_received = True
if not self.eval_response_received_first:
self.eval_response_received_first = True
self.get_logger().info('/active_rl_eval_response connected!')
else:
self.eval_response = response
self.eval_response_received = True
def active_rl_callback(self, request, response):
@ -86,7 +92,7 @@ class ActiveRLService(Node):
self.eval_pub.publish(eval_request)
while not self.eval_response_received:
rclpy.spin_once(self)
time.sleep(0.1)
self.get_logger().info('Topic responded!')
new_policy = self.eval_response.policy