dataloader kwargs part removed

snehilchatterjee · snehilchatterjee · commit 0610bd3cb38f · 2024-10-08T21:46:48.000+05:30
diff --git a/src/pytorch_tabular/tabular_datamodule.py b/src/pytorch_tabular/tabular_datamodule.py
@@ -809,7 +809,6 @@ def train_dataloader(self, batch_size: Optional[int] = None) -> DataLoader:
             num_workers=self.config.num_workers,
             sampler=self.train_sampler,
             pin_memory=self.config.pin_memory,
-            **self.config.dataloader_kwargs,
         )
 
     def val_dataloader(self, batch_size: Optional[int] = None) -> DataLoader:
@@ -828,7 +827,6 @@ def val_dataloader(self, batch_size: Optional[int] = None) -> DataLoader:
             shuffle=False,
             num_workers=self.config.num_workers,
             pin_memory=self.config.pin_memory,
-            **self.config.dataloader_kwargs,
         )
 
     def _prepare_inference_data(self, df: DataFrame) -> DataFrame:
@@ -871,7 +869,6 @@ def prepare_inference_dataloader(
             batch_size or self.batch_size,
             shuffle=False,
             num_workers=self.config.num_workers,
-            **self.config.dataloader_kwargs,
         )
 
     def save_dataloader(self, path: Union[str, Path]) -> None:

Original file line number	Diff line number	Diff line change
`@@ -809,7 +809,6 @@ def train_dataloader(self, batch_size: Optional[int] = None) -> DataLoader:`
`809`	`809`	`num_workers=self.config.num_workers,`
`810`	`810`	`sampler=self.train_sampler,`
`811`	`811`	`pin_memory=self.config.pin_memory,`
`812`		`- **self.config.dataloader_kwargs,`
`813`	`812`	`)`
`814`	`813`
`815`	`814`	`def val_dataloader(self, batch_size: Optional[int] = None) -> DataLoader:`
`@@ -828,7 +827,6 @@ def val_dataloader(self, batch_size: Optional[int] = None) -> DataLoader:`
`828`	`827`	`shuffle=False,`
`829`	`828`	`num_workers=self.config.num_workers,`
`830`	`829`	`pin_memory=self.config.pin_memory,`
`831`		`- **self.config.dataloader_kwargs,`
`832`	`830`	`)`
`833`	`831`
`834`	`832`	`def _prepare_inference_data(self, df: DataFrame) -> DataFrame:`
`@@ -871,7 +869,6 @@ def prepare_inference_dataloader(`
`871`	`869`	`batch_size or self.batch_size,`
`872`	`870`	`shuffle=False,`
`873`	`871`	`num_workers=self.config.num_workers,`
`874`		`- **self.config.dataloader_kwargs,`
`875`	`872`	`)`
`876`	`873`
`877`	`874`	`def save_dataloader(self, path: Union[str, Path]) -> None:`