hotfix for predictions plotting

UMDBPP · Mar 10, 2021 · 945b566 · 945b566
1 parent 0c383e1
commit 945b566
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/packetraven/plotting.py b/packetraven/plotting.py
@@ -34,7 +34,7 @@ def __init__(
  raise NotImplementedError(f'unsupported plotting variable "{variable}"')
 
  self.packet_tracks = packet_tracks
- self.predictions = predictions
+ self.predictions = predictions if predictions is not None else {}
  self.variable = variable
 
  self.window.protocol('WM_DELETE_WINDOW', self.window.iconify)
@@ -48,7 +48,7 @@ def update(
  ):
  if packet_tracks is not None:
  self.packet_tracks.update(packet_tracks)
- if packet_tracks is not None:
+ if predictions is not None:
  self.predictions.update(predictions)
 
  if len(self.packet_tracks) > 0: