From 3e0eed7b4d051988e759e8b7ba3ad0a1fd8522aa Mon Sep 17 00:00:00 2001 From: Khaled Sulayman Date: Mon, 13 Jan 2025 21:46:57 +0400 Subject: [PATCH] delete unused arg taxonomy_path in leaf_node_to_samples Signed-off-by: Khaled Sulayman --- src/instructlab/sdg/generate_data.py | 1 - src/instructlab/sdg/utils/taxonomy.py | 2 -- 2 files changed, 3 deletions(-) diff --git a/src/instructlab/sdg/generate_data.py b/src/instructlab/sdg/generate_data.py index d97cdc27..30166763 100644 --- a/src/instructlab/sdg/generate_data.py +++ b/src/instructlab/sdg/generate_data.py @@ -416,7 +416,6 @@ def generate_data( leaf_node_path = leaf_node[0]["taxonomy_path"].replace("->", "_") samples = leaf_node_to_samples( leaf_node, - taxonomy, server_ctx_size, chunk_word_count, document_output_dir, diff --git a/src/instructlab/sdg/utils/taxonomy.py b/src/instructlab/sdg/utils/taxonomy.py index c0d4dc1f..cf8073c5 100644 --- a/src/instructlab/sdg/utils/taxonomy.py +++ b/src/instructlab/sdg/utils/taxonomy.py @@ -429,7 +429,6 @@ def map_chunks_to_icls(chunks: List, leaf_node: Dict) -> Dataset: def _knowledge_leaf_node_to_samples( leaf_node, - taxonomy_path, # pylint: disable=unused-argument server_ctx_size, chunk_word_count, document_output_dir, @@ -480,7 +479,6 @@ def leaf_node_to_samples( if leaf_node[0].get("documents"): return _knowledge_leaf_node_to_samples( leaf_node, - taxonomy_path, server_ctx_size, chunk_word_count, document_output_dir,