X-Git-Url: https://gerrit.fd.io/r/gitweb?p=csit.git;a=blobdiff_plain;f=resources%2Ftraffic_scripts%2Fipfix_sessions.py;h=e7597a894a3519fb68b0288e3936e90afccfaa23;hp=385d21362b353542e38353d22738b0aa737fbaf5;hb=d166ae0881c29dfd05ed61b9a12156f17981bb6d;hpb=2e070f982606a7846984d38a8b0f58c12df6824f

diff --git a/resources/traffic_scripts/ipfix_sessions.py b/resources/traffic_scripts/ipfix_sessions.py
index 385d21362b..e7597a894a 100755
--- a/resources/traffic_scripts/ipfix_sessions.py
+++ b/resources/traffic_scripts/ipfix_sessions.py
@@ -22,7 +22,8 @@ from scapy.layers.inet import IP, TCP, UDP
 from scapy.layers.inet6 import IPv6
 from scapy.layers.l2 import Ether
 
-from resources.libraries.python.IPFIXUtil import IPFIXHandler, IPFIXData
+from resources.libraries.python.telemetry.IPFIXUtil import IPFIXHandler, \
+    IPFIXData
 from resources.libraries.python.PacketVerifier import RxQueue, TxQueue, auto_pad
 from resources.libraries.python.TrafficScriptArg import TrafficScriptArg
 
@@ -133,8 +134,6 @@ def main():
     tx_if = args.get_arg('tx_if')
 
     protocol = args.get_arg('protocol')
-    source_port = int(args.get_arg('port'))
-    destination_port = int(args.get_arg('port'))
     count = int(args.get_arg('count'))
     sessions = int(args.get_arg('sessions'))
 
@@ -167,8 +166,8 @@ def main():
 
     # do not print details for sent packets
     verbose = False
-    print("Sending more than one packet. Details will be filtered for\
-    # all packets sent.")
+    print("Sending more than one packet. Details will be filtered for "
+          "all packets sent.")
 
     ignore = []
     for x in range(sessions):