Are you aware of the kedro-mlflow plugin created by our community? If the logic is only specific to a node, it may better be just included in the node's logic?