aeppl.rewriting.PreserveRVMappings.on_import#

PreserveRVMappings.on_import(fgraph, node, reason)#

Called whenever a node is imported into fgraph, which is just before the node is actually connected to the graph.

Note: this is not called when the graph is created. If you want to detect the first nodes to be implemented to the graph, you should do this by implementing on_attach.