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,