diff --git a/olaflow/processing/diff.py b/olaflow/processing/diff.py index 6f41763..c8788b1 100644 --- a/olaflow/processing/diff.py +++ b/olaflow/processing/diff.py @@ -57,8 +57,8 @@ def get_pickle(out): models = list(map(get_pickle, args.output)) +fig, ax = plt.subplots(len(models), figsize=(6, 1.5 * len(models)), dpi=100, constrained_layout=True) if args.timestep is None: - fig, ax = plt.subplots(len(models), constrained_layout=True) for i, (_ax, _model) in enumerate(zip(ax, models)): _ax.contour( _model.t, @@ -77,7 +77,6 @@ if args.timestep is None: ) ) else: - fig, ax = plt.subplots(len(models), constrained_layout=True) for i, (_ax, _model) in enumerate(zip(ax, models)): _ax.tricontour( _model.x,