diff --git a/tests/unit/test_decoder_encoder.py b/tests/unit/test_decoder_encoder.py index 41fabf5..5d06f69 100644 --- a/tests/unit/test_decoder_encoder.py +++ b/tests/unit/test_decoder_encoder.py @@ -44,8 +44,9 @@ def test_decode_csv(): {"question": "where is Berlin?", "context": "Berlin is the capital of Germany"}, ] } + text_classification_input = "inputs\r\nI love you\r\nI like you" - decoded_data = decoder_encoder.decode_csv(DECODE_CSV_INPUT) + decoded_data = decoder_encoder.decode_csv(text_classification_input) assert decoded_data == {"inputs": ["I love you", "I like you"]}