Skip to content

Commit 25b0137

Browse files
authored
update base_model.py (#143)
1 parent 7a48a2b commit 25b0137

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/pytorch_tabular/models/base_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -123,13 +123,13 @@ def __init__(
123123
)
124124
if not WANDB_INSTALLED:
125125
self.do_log_logits = False
126-
raise warnings.warn(
126+
warnings.warn(
127127
"Wandb is not installed. Please install wandb to log logits. "
128128
"You can install wandb using pip install wandb or install PyTorch Tabular using pip install pytorch-tabular[all]"
129129
)
130130
if not PLOTLY_INSTALLED:
131131
self.do_log_logits = False
132-
raise warnings.warn(
132+
warnings.warn(
133133
"Plotly is not installed. Please install plotly to log logits. "
134134
"You can install plotly using pip install plotly or install PyTorch Tabular using pip install pytorch-tabular[all]"
135135
)

0 commit comments

Comments
 (0)