Skip to content

Commit 84e3522

Browse files
committed
Fix for coding style
1 parent 5294b3f commit 84e3522

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

art/attacks/evasion/steal_now_attack_later/drop_block2d.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ def drop_block2d(x: "torch.Tensor", prob: float, block_size: int):
5858
N, _, H, W = x.size() # pylint: disable=C0103
5959
block_size = min(block_size, W, H)
6060
# compute the gamma of Bernoulli distribution
61-
gamma = (prob * H * W) / ((block_size ** 2) * ((H - block_size + 1) * (W - block_size + 1)))
61+
gamma = (prob * H * W) / ((block_size**2) * ((H - block_size + 1) * (W - block_size + 1)))
6262
noise = torch.empty((N, 1, H - block_size + 1, W - block_size + 1), dtype=x.dtype, device=x.device)
6363
noise.bernoulli_(gamma)
6464

tests/attacks/evasion/test_steal_now_attack_later.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ def collect_patches_from_images(model, imgs):
116116
f.write(response.content)
117117
time.sleep(0.5)
118118

119-
#% Collect patches
119+
# % Collect patches
120120
import glob
121121
from torchvision import transforms
122122
from torchvision.datasets.vision import VisionDataset

0 commit comments

Comments
 (0)