Import yaml

main.py

@@ -3,6 +3,7 @@ import sys
 import random
 
 import simpy
+import yaml
 
 from pydyton import Network, NodeFactory
 from pydyton.traces import types as traces