Skip to content

Commit b8d30cf

Browse files
committed
Fix ruff
1 parent 6ae8350 commit b8d30cf

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

delphi/utils.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,11 +48,15 @@ def load_tokenized_data(
4848

4949
try:
5050
from datasets import Column
51+
5152
if isinstance(tokens, Column):
5253
from datasets.table import table_iter
54+
5355
tokens = torch.cat(
5456
[
55-
torch.from_numpy(np.stack(table_chunk["input_ids"].to_numpy(), axis=0))
57+
torch.from_numpy(
58+
np.stack(table_chunk["input_ids"].to_numpy(), axis=0)
59+
)
5660
for table_chunk in table_iter(
5761
tokens.source._data, convert_to_tensor_chunk_size
5862
)

0 commit comments

Comments
 (0)