import factory

from .. import models


class NodeFactory(factory.DjangoModelFactory):
    FACTORY_FOR = models.Node

import factory

from .. import models


class NodeFactory(factory.DjangoModelFactory):
    class Meta:
        model = models.Node

Use new format for factory boy factories