docs, tests: replace datasets.rst with memory_optimizations.rst

datasets.rst was removed from torchtune repo.

Signed-off-by: Ihar Hrachyshka <ihar.hrachyshka@gmail.com>
This commit is contained in:
Ihar Hrachyshka 2025-02-05 10:21:44 -05:00
parent 529708215c
commit 8f9b8a57cb
5 changed files with 3 additions and 5 deletions

View file

@ -1662,7 +1662,7 @@
"from termcolor import cprint\n",
"from llama_stack_client.types import Document\n",
"\n",
"urls = [\"chat.rst\", \"llama3.rst\", \"datasets.rst\", \"lora_finetune.rst\"]\n",
"urls = [\"chat.rst\", \"llama3.rst\", \"memory_optimizations.rst\", \"lora_finetune.rst\"]\n",
"documents = [\n",
" Document(\n",
" document_id=f\"num-{i}\",\n",

View file

@ -185,7 +185,7 @@ client = (
) # or create_http_client() depending on the environment you picked
# Documents to be used for RAG
urls = ["chat.rst", "llama3.rst", "datasets.rst", "lora_finetune.rst"]
urls = ["chat.rst", "llama3.rst", "memory_optimizations.rst", "lora_finetune.rst"]
documents = [
Document(
document_id=f"num-{i}",

View file

@ -152,7 +152,6 @@ class TestAgents:
"memory_optimizations.rst",
"chat.rst",
"llama3.rst",
"datasets.rst",
"qat_finetune.rst",
"lora_finetune.rst",
]

View file

@ -28,7 +28,6 @@ def sample_documents():
"memory_optimizations.rst",
"chat.rst",
"llama3.rst",
"datasets.rst",
"qat_finetune.rst",
"lora_finetune.rst",
]

View file

@ -264,7 +264,7 @@ def test_custom_tool(llama_stack_client, agent_config):
def test_rag_agent(llama_stack_client, agent_config):
urls = ["chat.rst", "llama3.rst", "datasets.rst", "lora_finetune.rst"]
urls = ["chat.rst", "llama3.rst", "memory_optimizations.rst", "lora_finetune.rst"]
documents = [
Document(
document_id=f"num-{i}",