From 93fd83c967567eae5e57ca32e073cd2e09d2e756 Mon Sep 17 00:00:00 2001 From: Manu Joseph Date: Sun, 26 Nov 2023 10:15:35 +0530 Subject: [PATCH] fixing assert statement --- src/pytorch_tabular/tabular_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/pytorch_tabular/tabular_model.py b/src/pytorch_tabular/tabular_model.py index 1875b905..1f1ded60 100644 --- a/src/pytorch_tabular/tabular_model.py +++ b/src/pytorch_tabular/tabular_model.py @@ -664,7 +664,7 @@ def fit( ), "`fit` is not valid for SSL task. Please use `pretrain` for semi-supervised learning" if metrics is not None: assert len(metrics) == len( - metrics_prob_inputs + metrics_prob_inputs or [] ), "The length of `metrics` and `metrics_prob_inputs` should be equal" seed = seed or self.config.seed if seed: