Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit 1c9064a

Browse files
committedApr 15, 2024·
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent f8a5859 commit 1c9064a

File tree

2 files changed

+15
-11
lines changed

2 files changed

+15
-11
lines changed
 

‎dataset_loaders/dataset_loaders.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -887,9 +887,11 @@ def __init__(
887887

888888
# Allow up to one level of nesting
889889
self.feats = [
890-
[load_features(df, feats_inner) for feats_inner in feats]
891-
if isinstance(feats, list)
892-
else load_features(df, feats)
890+
(
891+
[load_features(df, feats_inner) for feats_inner in feats]
892+
if isinstance(feats, list)
893+
else load_features(df, feats)
894+
)
893895
for feats in input_features
894896
]
895897

‎evaluation/retrieval_evaluation.py

+10-8
Original file line numberDiff line numberDiff line change
@@ -179,14 +179,16 @@ def retrieval_evaluation(
179179
frames = frames[:, ::frame_stride]
180180
splits = torch.split(frames, nframes, 1)
181181
splits_pad = [
182-
x
183-
if x.shape[1] == nframes
184-
else torch.index_select(
185-
x,
186-
dim=1,
187-
index=torch.floor(
188-
torch.linspace(0, x.shape[1] - 1, nframes, device=device)
189-
).to(torch.int64),
182+
(
183+
x
184+
if x.shape[1] == nframes
185+
else torch.index_select(
186+
x,
187+
dim=1,
188+
index=torch.floor(
189+
torch.linspace(0, x.shape[1] - 1, nframes, device=device)
190+
).to(torch.int64),
191+
)
190192
)
191193
for x in splits
192194
]

0 commit comments

Comments
 (0)
Please sign in to comment.