Skip to content

Commit 1b62eee

Browse files
micedremeilame-tayebjee
authored andcommitted
Fix check for categorical variable
1 parent 89cc8fe commit 1b62eee

1 file changed

Lines changed: 0 additions & 6 deletions

File tree

torchTextClassifiers/torchTextClassifiers.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -223,12 +223,6 @@ def train(
223223
X_train, y_train = self._check_XY(X_train, y_train)
224224
X_val, y_val = self._check_XY(X_val, y_val)
225225

226-
assert (
227-
X_train["categorical_variables"].ndim > 1
228-
and X_train["categorical_variables"].shape[1] == X_val["categorical_variables"].shape[1]
229-
or X_val["categorical_variables"].ndim == 1
230-
)
231-
232226
if (
233227
X_train["categorical_variables"] is not None
234228
and X_val["categorical_variables"] is not None

0 commit comments

Comments
 (0)