Skip to content

Commit 4a25899

Browse files
committed
Fix sklearn test
1 parent 1d0c498 commit 4a25899

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

cebra/solver/base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -639,8 +639,8 @@ def transform(self,
639639

640640
model, offset = self._select_model(inputs, session_id)
641641

642-
#if len(offset) < 2 and pad_before_transform:
643-
# pad_before_transform = False
642+
if len(offset) < 2 and pad_before_transform:
643+
pad_before_transform = False
644644

645645
model.eval()
646646
if batch_size is not None and inputs.shape[0] > int(

0 commit comments

Comments
 (0)