Изменения падежа, рода и числа числительного после pytils

import nltk
import os
import pymorphy2
import string
 
nltk.download('punkt')
nltk.download('averaged_perceptron_tagger')
nltk.download('universal_tagset')
 
def fix_numeral_gender_case(text):
    morph = pymorphy2.MorphAnalyzer()
    # Разбиваем текст на слова
    words = nltk.word_tokenize(text)
    # Получаем тэги частей речи для каждого слова
    tagged_words = nltk.pos_tag(words, tagset='universal')
    # Создаем словарь для соответствия тэгов и падежей
    tag_to_case = {'NOUN': 'nominative', 'NUM': 'accusative', '.': None}
    # Создаем словарь для соответствия тэгов и родов
    tag_to_gender = {'NOUN': None, 'NUM': None, '.': None}
    # Проходим по каждому слову в тексте
    for i, (word, tag) in enumerate(tagged_words):
        # Проверяем, является ли слово числительным и нуждается ли оно в изменении падежа, рода или числа
        if tag == 'NUM':
            # Определяем нужный падеж для числительного слова
            case = tag_to_case[tagged_words[i-1][1]]
            # Определяем нужный род для числительного слова
            gender = tag_to_gender[tagged_words[i-1][1]]
            if gender is None:
                # Определяем род числительного слова
                parsed_word = morph.parse(word)[0]
                gender = parsed_word.tag.gender or 'neuter'
            # Получаем измененную форму числительного слова
            modified_word = parsed_word.inflect({case, gender}).word
            # Заменяем числительное слово в тексте на измененную форму
            words[i] = modified_word
    # Объединяем слова обратно в текст
    return ' '.join(words)
 
 
text = "Я купил 3 карандаша и заплатил 120 рублей. 14 летний мальчик пришел 9 в класс под номером 3"
fixed_text = fix_numeral_gender_case(text)
print(fixed_text)

tag_to_case = {
    'ADJ': 'nominative',
    'ADP': None,
    'ADV': None,
    'CONJ': None,
    'DET': None,
    'NOUN': 'nominative',
    'NUM': 'accusative',
    'PRT': None,
    'PRON': None,
    'VERB': None,
    '.': None
}
 
tag_to_gender = {
    'ADJ': 'masc',
    'ADP': None,
    'ADV': None,
    'CONJ': None,
    'DET': None,
    'NOUN': None,
    'NUM': None,
    'PRT': None,
    'PRON': None,
    'VERB': 'masc',
    '.': None
}