From 57ca51a0a10878f36903717ab895fb851dc06a93 Mon Sep 17 00:00:00 2001
From: Daiki Katsuragawa <50144563+daikikatsuragawa@users.noreply.github.com>
Date: Sun, 13 Nov 2022 17:47:01 +0000
Subject: [PATCH] Fix type hint

Signed-off-by: Daiki Katsuragawa <50144563+daikikatsuragawa@users.noreply.github.com>
---
 obp/policy/offline_continuous.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/obp/policy/offline_continuous.py b/obp/policy/offline_continuous.py
index 8ce6a697..0324293d 100644
--- a/obp/policy/offline_continuous.py
+++ b/obp/policy/offline_continuous.py
@@ -526,7 +526,7 @@ def _estimate_policy_gradient(
         reward: torch.Tensor,
         pscore: torch.Tensor,
         action_by_current_policy: torch.Tensor,
-    ) -> float:
+    ) -> torch.Tensor:
         """Estimate the policy gradient.
 
         Parameters