Skip to content

Commit 312a1fe

Browse files
committed
update for bandit
1 parent 51b504c commit 312a1fe

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

raft_toolkit/core/formatters/dataset_converter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -346,7 +346,7 @@ def main():
346346

347347
logger = logging.getLogger("raft")
348348
# Load dataset from local files only - not from HuggingFace Hub
349-
ds = load_dataset(input_type, data_files={"train": args.input}, trust_remote_code=False)["train"]
349+
ds = load_dataset(input_type, data_files={"train": args.input}, trust_remote_code=False)["train"] # nosec B615
350350
logger.info(f"Dataset has {ds.num_rows} rows")
351351
formatter = DatasetConverter()
352352

0 commit comments

Comments
 (0)