IBM / unitxt / 21828313409

81%

push

github

Fix qa evaluation data classification policy (#1962)

* Fix data classification policy warning in qa_evaluation.py

- 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>

* Replace deprecated torch_dtype with dtype parameter

- 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>

---------

Signed-off-by: Yoav Katz <katz@il.ibm.com>

1607 of 2006 branches covered (80.11%)

Branch coverage included in aggregate %.

10954 of 13520 relevant lines covered (81.02%)

0.81 hits per line