Skip to content

Commit 90818aa

Browse files
committed
Rename dataset to 0din_threatfeed to distinguish from existing 0din_* n-day datasets
PR microsoft#1398 already added six static, hand-curated 0din_* local n-day disclosure datasets. This live API loader pulls the gated 0DIN threat feed, so name it 0din_threatfeed to clearly distinguish the dynamic full feed from the static disclosures and parallel the existing 0din_* naming. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
1 parent e69f2b6 commit 90818aa

3 files changed

Lines changed: 5 additions & 5 deletions

File tree

doc/code/datasets/1_loading_datasets.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,12 +70,12 @@
7070
"data": {
7171
"text/plain": [
7272
"['0din_chemical_compiler_debug',\n",
73-
" '0din',\n",
7473
" '0din_correction',\n",
7574
" '0din_hex_recipe_book',\n",
7675
" '0din_incremental_table_completion',\n",
7776
" '0din_placeholder_injection',\n",
7877
" '0din_technical_field_guide',\n",
78+
" '0din_threatfeed',\n",
7979
" 'adv_bench',\n",
8080
" 'aegis_content_safety',\n",
8181
" 'agent_threat_rules',\n",

pyrit/datasets/seed_datasets/remote/odin_dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ def __init__(
160160
@override
161161
def dataset_name(self) -> str:
162162
"""Return the dataset name."""
163-
return "0din"
163+
return "0din_threatfeed"
164164

165165
def _resolve_api_key(self) -> str:
166166
"""

tests/unit/datasets/test_odin_dataset.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@ class TestODINDatasetInit:
125125

126126
def test_init_with_api_key(self, api_key):
127127
loader = _ODINDataset(api_key=api_key)
128-
assert loader.dataset_name == "0din"
128+
assert loader.dataset_name == "0din_threatfeed"
129129
assert loader._api_key == api_key
130130

131131
def test_init_with_env_var(self, api_key):
@@ -169,7 +169,7 @@ def test_init_accepts_valid_enums(self, api_key):
169169
assert loader._categories == [ODINTaxonomyCategory.STRATAGEMS, ODINTaxonomyCategory.LANGUAGE]
170170

171171
def test_dataset_name(self, api_key):
172-
assert _ODINDataset(api_key=api_key).dataset_name == "0din"
172+
assert _ODINDataset(api_key=api_key).dataset_name == "0din_threatfeed"
173173

174174

175175
class TestODINDatasetFetch:
@@ -216,7 +216,7 @@ async def test_seed_prompt_fields(self, api_key, single_page_response):
216216
prompts = [s for s in dataset.seeds if isinstance(s, SeedPrompt)]
217217
first = prompts[0]
218218
assert first.data_type == "text"
219-
assert first.dataset_name == "0din"
219+
assert first.dataset_name == "0din_threatfeed"
220220
assert first.name == "Report A"
221221
assert first.description == "A short summary."
222222
assert first.harm_categories == ["language", "stratagems"]

0 commit comments

Comments
 (0)