Fix missing import

stefanDeveloper · Sep 15, 2023 · 4b0860f · 4b0860f
1 parent 360f4dd
commit 4b0860f
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/heifip/cli.py b/heifip/cli.py
@@ -6,13 +6,11 @@
     )
 
 from heifip import CONTEXT_SETTINGS, __version__
-from heifip.images import NetworkTrafficImage
 from heifip.images.flow import FlowImage
 from heifip.images.flow_tiled_auto import FlowImageTiledAuto
 from heifip.images.flow_tiled_fixed import FlowImageTiledFixed
-from heifip.images.markovchain import (MarkovTransitionMatrixFlow,
-                                       MarkovTransitionMatrixPacket)
-from heifip.images.packet import Packet
+from heifip.images.markovchain import (MarkovTransitionMatrixFlow)
+from heifip.images.packet import PacketImage
 from heifip.layers import PacketProcessorType
 from heifip.main import Runner