From fd594305ede4a4376052e3a20d7f089168dc1ff9 Mon Sep 17 00:00:00 2001 From: jameschapman19 Date: Mon, 4 Sep 2023 15:27:15 +0000 Subject: [PATCH] Format code with black --- proxtorch/operators/tvl1_3d.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/proxtorch/operators/tvl1_3d.py b/proxtorch/operators/tvl1_3d.py index f80d46a..d26e0e5 100644 --- a/proxtorch/operators/tvl1_3d.py +++ b/proxtorch/operators/tvl1_3d.py @@ -76,7 +76,7 @@ def gradient(self, x): gradients[d, ...] = F.pad( torch.diff(x, dim=d, n=1), pad=get_padding_tuple(d, x.dim()) ) - gradients[:-1] *= (1.0 - self.l1_ratio) + gradients[:-1] *= 1.0 - self.l1_ratio gradients[-1] = self.l1_ratio * x return gradients