Skip to content

Commit cd3dec0

Browse files
committed
Fix linter errors
1 parent 8e661df commit cd3dec0

File tree

2 files changed

+10
-5
lines changed

2 files changed

+10
-5
lines changed

noxfile.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,8 @@ def lint(session):
1616
@nox.session(python=["3.8", "3.9", "3.10", "3.11", "3.12"])
1717
def tests(session):
1818
session.install(
19-
'torch==2.2.1',
20-
'torchvision',
19+
'torch==2.2.1',
20+
'torchvision',
2121
'--index-url', 'https://download.pytorch.org/whl/cpu'
2222
)
2323
session.install('.')

src/pytorch_fid/fid_score.py

+8-3
Original file line numberDiff line numberDiff line change
@@ -64,8 +64,9 @@ def tqdm(x):
6464
help=('Dimensionality of Inception features to use. '
6565
'By default, uses pool3 features'))
6666
parser.add_argument('--save-stats', action='store_true',
67-
help=('Generate an npz archive from a directory of samples. '
68-
'The first path is used as input and the second as output.'))
67+
help=('Generate an npz archive from a directory of '
68+
'samples. The first path is used as input and the '
69+
'second as output.'))
6970
parser.add_argument('path', type=str, nargs=2,
7071
help=('Paths to the generated images or '
7172
'to .npz statistic files'))
@@ -307,7 +308,11 @@ def main():
307308
num_workers = args.num_workers
308309

309310
if args.save_stats:
310-
save_fid_stats(args.path, args.batch_size, device, args.dims, num_workers)
311+
save_fid_stats(args.path,
312+
args.batch_size,
313+
device,
314+
args.dims,
315+
num_workers)
311316
return
312317

313318
fid_value = calculate_fid_given_paths(args.path,

0 commit comments

Comments
 (0)