CalvinBackup/agentic_security
1
0
Fork 0
mirror of https://github.com/msoedov/agentic_security.git synced 2026-06-24 14:19:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix(fmt):

Browse Source
This commit is contained in:
Alexander Myasoedov
2024-04-25 21:35:14 +03:00
parent 83b7c298cc
commit 4f4da2f136
4 changed files with 10 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
langalf/probe_data/data.py
+2 -3
View File
@@ -296,8 +296,8 @@ def load_local_csv() -> ProbeDataset:
def dataset_from_iterator(name: str, iterator) -> list:
"""
Convert an iterator into a list of prompts and create a ProbeDataset object.
"""Convert an iterator into a list of prompts and create a ProbeDataset
object.
Args:
name (str): The name of the dataset.
@@ -305,7 +305,6 @@ def dataset_from_iterator(name: str, iterator) -> list:
Returns:
list: A list containing a single ProbeDataset object.
"""
prompts = list(iterator)
tokens = count_words_in_list(prompts)