Skip to content

Commit 8ea9ff0

Browse files
committed
fix code style
1 parent b3c393d commit 8ea9ff0

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

DPF/filters/videos/videohash_filter.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
1-
import io
2-
from typing import Any
1+
import os
32
import shutil
4-
from videohash import VideoHash # type: ignore
3+
from typing import Any
54
from uuid import uuid4
6-
import os
5+
6+
from videohash import VideoHash # type: ignore
77

88
from DPF.types import ModalityToDataMapping
99

@@ -23,7 +23,7 @@ def __init__(
2323

2424
@property
2525
def result_columns(self) -> list[str]:
26-
return [f"video_hash"]
26+
return ["video_hash"]
2727

2828
@property
2929
def dataloader_kwargs(self) -> dict[str, Any]:
@@ -48,7 +48,7 @@ def preprocess_data(
4848
video_path = os.path.join(tmp_dir, 'video.mp4')
4949
with open(video_path, 'wb') as f:
5050
f.write(video_file)
51-
51+
5252
hash_obj = VideoHash(path=video_path, storage_path=tmp_dir)
5353
shutil.rmtree(hash_obj.storage_path)
5454
os.remove(video_path)

0 commit comments

Comments
 (0)