Update Modules.py

Fix bug caused by newer pytorch version

Author: ruochiz

Code/Modules.py

@@ -337,7 +337,7 @@ def __init__(
 		self.w_stack = []
 		self.dims = dims
 		for i in range(len(dims) - 1):
-			self.w_stack.append(nn.Conv1d(dims[i], dims[i + 1], 1, use_bias))
+			self.w_stack.append(nn.Conv1d(dims[i], dims[i + 1], 1, bias=use_bias))
 			self.add_module("PWF_Conv%d" % (i), self.w_stack[-1])
 		self.reshape = reshape
 		self.layer_norm = nn.LayerNorm(dims[-1])
@@ -689,4 +689,4 @@ def balance_num(self, edges):
 		for c in tqdm(choice):
 			final.append(np.random.choice(np.where(cell == c)[0], num, replace=True))
 		final = np.concatenate(final, axis=-1)
-		return final
+		return final