Skip to content

Conversation

@yoavkatz
Copy link
Member

  • Fixed data classification policy warning (examples/qa_evaluation.py):

Added data_classification_policy=["public"] parameter to create_dataset()
Removes warning: "LoadFromDictionary sets 'data_classification_policy' to ['proprietary'] by default..."

  • Fixed torch_dtype deprecation warning (src/unitxt/inference.py):

Replaced all torch_dtype parameters with dtype in model initialization calls
Updated in 4 locations across HFPipelineBasedInferenceEngine, LlavaInferenceEngine, and HFPeftInferenceEngine
Removes warning: "torch_dtype is deprecated! Use dtype instead!"
Verification:

Ran examples/qa_evaluation.py successfully with no warnings
All pre-commit checks passed (ruff, ruff-format, detect secrets, codespell, import enforcement)
Changes pushed to remote branch

- Add explicit data_classification_policy=['public'] parameter to create_dataset()
- Removes warning: 'LoadFromDictionary sets data_classification_policy to [proprietary] by default'

Signed-off-by: Yoav Katz <katz@il.ibm.com>
- Update all from_pretrained() calls to use 'dtype' instead of 'torch_dtype'
- Fixes deprecation warning from transformers library
- Changes in HFPipelineBasedInferenceEngine, LlavaInferenceEngine, and HFPeftInferenceEngine

Signed-off-by: Yoav Katz <katz@il.ibm.com>
@yoavkatz yoavkatz force-pushed the fix-qa-evaluation-data-classification-policy branch from 0b9f317 to 3d3b21d Compare January 18, 2026 09:55
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants