remove warning notice due to 1d structure

Summary: remove warning due to 1d structure of reward module. Reviewed By: danielrjiang, alexnikulkov Differential Revision: D53273291 fbshipit-source-id: 0d7e4ccafbbcde8c98a9804c5e4c6eb722a1c7fd
facebookresearch · Jan 31, 2024 · 322992e · 322992e
1 parent d86e29c
commit 322992e
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/pearl/neural_networks/contextual_bandit/linear_regression.py b/pearl/neural_networks/contextual_bandit/linear_regression.py
@@ -112,7 +112,6 @@ def _validate_train_inputs(
         if weight is None:
             weight = torch.ones_like(y)
         if weight.ndim == 1:
-            logger.warning("2D shape expected for weight, got 1D shape {weight.shape}")
             weight = weight.unsqueeze(-1)
         assert x.shape == (
             batch_size,