From 5dfdcfe5e31c022816e1d93fdc1f6bd1848b2d49 Mon Sep 17 00:00:00 2001
From: Chunhui Yang <alexyang9631@gmail.com>
Date: Wed, 12 Jul 2023 17:06:50 -0400
Subject: [PATCH] remove sampler for valid dataloader in pretrain

---
 pytorch_tabnet/pretraining_utils.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/pytorch_tabnet/pretraining_utils.py b/pytorch_tabnet/pretraining_utils.py
index 26d37a20..9afc3caf 100644
--- a/pytorch_tabnet/pretraining_utils.py
+++ b/pytorch_tabnet/pretraining_utils.py
@@ -59,7 +59,6 @@ def create_dataloaders(
             DataLoader(
                 PredictDataset(X),
                 batch_size=batch_size,
-                sampler=sampler,
                 shuffle=need_shuffle,
                 num_workers=num_workers,
                 drop_last=drop_last,