mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-06-29 03:14:19 +00:00
e8112b31ab
# What does this PR do? - move folder ## Test Plan **Unit Test** ``` pytest -v -s -m "huggingface" datasetio/test_datasetio.py ``` **E2E** ``` llama stack run ``` ``` llama-stack-client eval run_benchmark meta-reference-mmlu --num-examples 5 --output-dir ./ --eval-task-config ~/eval_task_config.json --visualize ``` <img width="657" alt="image" src="https://github.com/user-attachments/assets/63d53f9d-6c7e-4667-af8c-9d16c91ae6e3"> ## Before submitting - [ ] This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case). - [ ] Ran pre-commit to handle lint / formatting issues. - [ ] Read the [contributor guideline](https://github.com/meta-llama/llama-stack/blob/main/CONTRIBUTING.md), Pull Request section? - [ ] Updated relevant documentation. - [ ] Wrote necessary unit or integration tests.
17 lines
629 B
Python
17 lines
629 B
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the terms described in the LICENSE file in
|
|
# the root directory of this source tree.
|
|
from llama_stack.distribution.utils.config_dirs import RUNTIME_BASE_DIR
|
|
from llama_stack.providers.utils.kvstore.config import (
|
|
KVStoreConfig,
|
|
SqliteKVStoreConfig,
|
|
)
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class HuggingfaceDatasetIOConfig(BaseModel):
|
|
kvstore: KVStoreConfig = SqliteKVStoreConfig(
|
|
db_path=(RUNTIME_BASE_DIR / "huggingface_datasetio.db").as_posix()
|
|
) # Uses SQLite config specific to HF storage
|