running updated pre-commit plugins

This commit is contained in:
Dan Saunders
2025-03-20 10:30:18 -04:00
committed by Wing Lian
parent aaa4571826
commit 21bac7ce1a
130 changed files with 420 additions and 291 deletions

View File

@@ -1,6 +1,7 @@
"""
tests for chat_template prompt strategy
"""
# pylint: disable=duplicate-code
import logging
import unittest

View File

@@ -1,6 +1,7 @@
"""
Test module for alpaca integration w chatml
"""
import pytest
from datasets import Dataset
from tokenizers import AddedToken

View File

@@ -1,6 +1,7 @@
"""
Tests for utils in axolotl.utils.chat_templates
"""
import unittest
import pytest

View File

@@ -920,9 +920,11 @@ class TestChatTemplateConfigurations:
)
variables = prompter.get_chat_template_msg_variables(
actual_jinja_template
if actual_jinja_template
else actual_tokenizer.get_chat_template(),
(
actual_jinja_template
if actual_jinja_template
else actual_tokenizer.get_chat_template()
),
"messages",
)

View File

@@ -1,6 +1,7 @@
"""
Tests for loading DPO preference datasets with chatml formatting
"""
import unittest
import pytest

View File

@@ -1,6 +1,7 @@
"""
tests for jinja_template_analyzer
"""
import logging
import pytest

View File

@@ -1,6 +1,7 @@
"""
Test module for raw i/o data for prompts
"""
import pytest
from datasets import Dataset
from tokenizers import AddedToken