Есть ли способ получить список объектов Factory Boy subfactory с правильным типом объекта Django?

У меня есть фабрика-фабрика, которая использует Django. Мне нужно получить доступ к списку сгенерированных объектов из SubFactory. Однако, когда я пытаюсь получить этот список, я получаю TypeErrors, в которых говорится, что мои сгенерированные объекты "SubFactory", а не тип объекта, который мне на самом деле нужен. Есть идеи, что я упускаю?

# factories.py
class PartNoRelationFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = PartNoRelation

    id = factory.Sequence(lambda n: n + 1)

    parent_part_no = factory.SubFactory(PART_NO_FACTORY)
    child_part_no = factory.SubFactory(PART_NO_FACTORY)

# test_factories.py
def mock_child_part_nos(arg_1: PartNo) -> 'list[PartNo]':
    mock_part_no_relation_one = PartNoRelationFactory()
    mock_part_no_relation_two = PartNoRelationFactory()
    mock_part_no_relation_three = PartNoRelationFactory()

    return [
        mock_part_no_relation_one.child_part_no,
        mock_part_no_relation_two.child_part_no,
        mock_part_no_relation_three.child_part_no
        ]
    # error: "Expression of type "list[SubFactory]" cannot be assigned to return type "list[PartNo]"

Вернуться на верх