From 686f9137c0abf973c7d19fd8dc1f3123563900f9 Mon Sep 17 00:00:00 2001 From: Sebastiaan Huber Date: Wed, 24 Apr 2024 15:25:33 +0200 Subject: [PATCH] `BandsData`: Use f-strings in `_prepare_gnuplot` The `flynt` autoformatter was trying to correct them automatically but it would result in incorrect syntax, so the transformation is done manually. --- src/aiida/orm/nodes/data/array/bands.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/src/aiida/orm/nodes/data/array/bands.py b/src/aiida/orm/nodes/data/array/bands.py index b31f0f44d2..a33f104111 100644 --- a/src/aiida/orm/nodes/data/array/bands.py +++ b/src/aiida/orm/nodes/data/array/bands.py @@ -1033,20 +1033,17 @@ def _prepare_gnuplot( script.append(f'set ylabel "Dispersion ({self.units})"') if title: - script.append('set title "{}"'.format(title.replace('"', '"'))) + script.append(f'set title "{title}"') # Plot, escaping filename + filename = os.path.basename(dat_filename) if len(x) > 1: script.append(f'set xrange [{x_min_lim}:{x_max_lim}]') script.append('set grid xtics lt 1 lc rgb "#888888"') - script.append('plot "{}" with l lc rgb "#000000"'.format(os.path.basename(dat_filename).replace('"', '"'))) + script.append(f'plot "{filename}" with l lc rgb "#000000"') else: script.append('set xrange [-1.0:1.0]') - script.append( - 'plot "{}" using ($1-0.25):($2):(0.5):(0) with vectors nohead lc rgb "#000000"'.format( - os.path.basename(dat_filename).replace('"', '"') - ) - ) + script.append(f'plot "{filename}" using ($1-0.25):($2):(0.5):(0) with vectors nohead lc rgb "#000000"') script_data = '\n'.join(script) + '\n' extra_files = {dat_filename: raw_data}