add in error message

SlokomManel · Sep 5, 2021 · 8578e0c · 8578e0c
1 parent 60da381
commit 8578e0c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/obp/dataset/synthetic_continuous.py b/obp/dataset/synthetic_continuous.py
@@ -243,7 +243,7 @@ def calc_ground_truth_policy_value(
         check_array(array=action, name="action", expected_dim=1)
         if context.shape[1] != self.dim_context:
             raise ValueError(
-                "Expected `context.shape[1] == self.dim_context`, found it False"
+                "Expected `context.shape[1] == self.dim_context`, but found it False"
             )
         if context.shape[0] != action.shape[0]:
             raise ValueError(

diff --git a/obp/dataset/synthetic_slate.py b/obp/dataset/synthetic_slate.py
@@ -877,7 +877,7 @@ def calc_ground_truth_policy_value(
             )
         if context.shape[1] != self.dim_context:
             raise ValueError(
-                "Expected `context.shape[1] == self.dim_context`, found it False"
+                "Expected `context.shape[1] == self.dim_context`, but found it False"
             )
         if evaluation_policy_logit_.shape[0] != context.shape[0]:
             raise ValueError(