diff --git a/Overlap/plot_components.py b/Overlap/plot_components.py index da77c21a4e6f75d54e0cc6ed59ca22cf4f692b6f..ef2dcd8e8e5b6d51b015e62e2cd2f5de9a6104ac 100755 --- a/Overlap/plot_components.py +++ b/Overlap/plot_components.py @@ -7,6 +7,7 @@ import numpy as np from matplotlib import patches import cartopy.crs as ccrs import report_graph +import sys color_iter = itertools.cycle(('#1f77b4', '#aec7e8', '#ff7f0e', '#ffbb78', '#2ca02c', '#98df8a', @@ -40,6 +41,8 @@ def is_node_type(G, n, label): only_date = True) == label[1] elif label[0] == "node": return n in label[1] + else: + sys.exit(f"is_node_type: bad label: {label}") def plot_nbunch(G, nbunch, color = '#1f78b4', label = None, ax = None): if ax is None: ax = plt.gca() @@ -96,7 +99,6 @@ def animate_nbunch(G, nbunch): if __name__ == "__main__": import argparse - import sys from os import path sys.path.append(path.join(sys.path[0], "../Common")) import util_eddies