From 1fe8136a99edcb364f3a354bd7ce47cc3fc963a5 Mon Sep 17 00:00:00 2001 From: Joshua Spear Date: Fri, 6 Sep 2024 20:43:48 +0100 Subject: [PATCH] bug fix with policy class when running on GPU --- src/offline_rl_ope/components/Policy.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/offline_rl_ope/components/Policy.py b/src/offline_rl_ope/components/Policy.py index e9973f2..d05154a 100644 --- a/src/offline_rl_ope/components/Policy.py +++ b/src/offline_rl_ope/components/Policy.py @@ -167,7 +167,7 @@ def __call__( action_prs = self.postproc_tens(p_return.action_prs) self.collect_res_fn(action_prs) self.collect_act_func(actions) - return p_return.action_prs + return action_prs class GreedyDeterministic(BasePolicy):