From ba10719520d6fbb906d84ac4a3e7374b3279b893 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Thu, 20 Oct 2022 21:03:38 +0530
Subject: [PATCH] add arg

---
 enhancer/data/dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/enhancer/data/dataset.py b/enhancer/data/dataset.py
index 8110a2a..06a6f67 100644
--- a/enhancer/data/dataset.py
+++ b/enhancer/data/dataset.py
@@ -185,7 +185,7 @@ class TaskDataset(pl.LightningDataModule):
         output["noisy"] = torch.stack(output["noisy"], dim=0)
         return output
 
-    def worker_init_fn(self):
+    def worker_init_fn(self, _):
         worker_info = torch.utils.data.get_worker_info()
         dataset = worker_info.dataset
         worker_id = worker_info.id