Update data.py

[v1docq]

Summary

Context

[docu-mentor]

👋 Hi, I'm @docu-mentor, an LLM-powered GitHub app
powered by Anyscale Endpoints
that gives you actionable feedback on your writing.

Simply create a new comment in this PR that says:

@docu-mentor run

and I will start my analysis. I only look at what you changed
in this PR. If you only want me to look at specific files or folders,
you can specify them like this:

@docu-mentor run doc/ README.md

In this example, I'll have a look at all files contained in the "doc/"
folder and the file "README.md". All good? Let's get started!

[github-actions]

All PEP8 errors has been fixed, thanks ❤️

Comment last updated at

[nicl-nno]

По возможности добавь ещё мини-тест, покрывающий этот кейс (например на упомянутом примере TSF)