black: reformat entire repository with black

1 files changed, 184 insertions, 161 deletions

diff --git a/var/spack/repos/builtin/packages/gptune/package.py b/var/spack/repos/builtin/packages/gptune/package.py
index dafae71ea3..9a29f34bbc 100644
--- a/var/spack/repos/builtin/packages/gptune/package.py
+++ b/var/spack/repos/builtin/packages/gptune/package.py
@@ -13,198 +13,221 @@ class Gptune(CMakePackage):
     Bayesian optimization methodologies."""
 
     homepage = "https://gptune.lbl.gov/"
-    url      = "https://github.com/gptune/GPTune/archive/refs/tags/3.0.0.tar.gz"
-    git      = "https://github.com/gptune/GPTune.git"
-    maintainers = ['liuyangzhuan']
-
-    version('master', branch='master')
-    version('3.0.0', sha256='e19bfc3033fff11ff8c20cae65b88b7ca005d2c4e4db047f9f23226126ec92fa')
-    version('2.1.0', sha256='737e0a1d83f66531098beafa73dd479f12def576be83b1c7b8ea5f1615d60a53')
-
-    variant('superlu', default=False, description='Build the SuperLU_DIST example')
-    variant('hypre', default=False, description='Build the Hypre example')
-    variant('mpispawn', default=True, description='MPI spawning-based interface')
-
-    depends_on('mpi', type=('build', 'link', 'run'))
-    depends_on('cmake@3.17:', type='build')
-    depends_on('jq', type='run')
-    depends_on('blas', type='link')
-    depends_on('lapack', type='link')
-    depends_on('scalapack', type='link')
-    depends_on('py-setuptools', type='build')
-    depends_on('py-ipyparallel', type=('build', 'run'))
-    depends_on('py-numpy@:1.21.5', type=('build', 'run'))
-    depends_on('py-pandas', type=('build', 'run'))
-    depends_on('py-joblib', type=('build', 'run'))
-    depends_on('py-scikit-learn', type=('build', 'run'))
-    depends_on('py-matplotlib', type=('build', 'run'))
-    depends_on('py-pyyaml', type=('build', 'run'))
-    depends_on('py-scikit-optimize@master+gptune', type=('build', 'run'))
-    depends_on('py-gpy', type=('build', 'run'))
-    depends_on('py-lhsmdu', type=('build', 'run'))
-    depends_on('py-hpbandster', type=('build', 'run'))
-    depends_on('py-opentuner', type=('build', 'run'))
-    depends_on('py-ytopt-autotune@1.1.0', type=('build', 'run'))
-    depends_on('py-filelock', type=('build', 'run'))
-    depends_on('py-requests', type=('build', 'run'))
-    depends_on('py-cython', type=('build', 'run'))
-    depends_on('py-pyaml', type=('build', 'run'))
-    depends_on('py-statsmodels@0.13.0:', type=('build', 'run'))
-    depends_on('py-mpi4py@3.0.3:', type=('build', 'run'))
-    depends_on('pygmo', type=('build', 'run'))
-    depends_on('openturns', type=('build', 'run'))
-    depends_on('py-pymoo', type=('build', 'run'), when='@3.0.0:')
-
-    depends_on('superlu-dist@develop', when='+superlu', type=('build', 'run'))
-    depends_on('hypre+gptune@2.19.0', when='+hypre', type=('build', 'run'))
-
-    depends_on('openmpi@4:', when='+mpispawn', type=('build', 'run'))
-    conflicts('mpich', when='+mpispawn')
-    conflicts('spectrum-mpi', when='+mpispawn')
-    conflicts('cray-mpich', when='+mpispawn')
-    conflicts('gcc@:7')
+    url = "https://github.com/gptune/GPTune/archive/refs/tags/3.0.0.tar.gz"
+    git = "https://github.com/gptune/GPTune.git"
+    maintainers = ["liuyangzhuan"]
+
+    version("master", branch="master")
+    version("3.0.0", sha256="e19bfc3033fff11ff8c20cae65b88b7ca005d2c4e4db047f9f23226126ec92fa")
+    version("2.1.0", sha256="737e0a1d83f66531098beafa73dd479f12def576be83b1c7b8ea5f1615d60a53")
+
+    variant("superlu", default=False, description="Build the SuperLU_DIST example")
+    variant("hypre", default=False, description="Build the Hypre example")
+    variant("mpispawn", default=True, description="MPI spawning-based interface")
+
+    depends_on("mpi", type=("build", "link", "run"))
+    depends_on("cmake@3.17:", type="build")
+    depends_on("jq", type="run")
+    depends_on("blas", type="link")
+    depends_on("lapack", type="link")
+    depends_on("scalapack", type="link")
+    depends_on("py-setuptools", type="build")
+    depends_on("py-ipyparallel", type=("build", "run"))
+    depends_on("py-numpy@:1.21.5", type=("build", "run"))
+    depends_on("py-pandas", type=("build", "run"))
+    depends_on("py-joblib", type=("build", "run"))
+    depends_on("py-scikit-learn", type=("build", "run"))
+    depends_on("py-matplotlib", type=("build", "run"))
+    depends_on("py-pyyaml", type=("build", "run"))
+    depends_on("py-scikit-optimize@master+gptune", type=("build", "run"))
+    depends_on("py-gpy", type=("build", "run"))
+    depends_on("py-lhsmdu", type=("build", "run"))
+    depends_on("py-hpbandster", type=("build", "run"))
+    depends_on("py-opentuner", type=("build", "run"))
+    depends_on("py-ytopt-autotune@1.1.0", type=("build", "run"))
+    depends_on("py-filelock", type=("build", "run"))
+    depends_on("py-requests", type=("build", "run"))
+    depends_on("py-cython", type=("build", "run"))
+    depends_on("py-pyaml", type=("build", "run"))
+    depends_on("py-statsmodels@0.13.0:", type=("build", "run"))
+    depends_on("py-mpi4py@3.0.3:", type=("build", "run"))
+    depends_on("pygmo", type=("build", "run"))
+    depends_on("openturns", type=("build", "run"))
+    depends_on("py-pymoo", type=("build", "run"), when="@3.0.0:")
+
+    depends_on("superlu-dist@develop", when="+superlu", type=("build", "run"))
+    depends_on("hypre+gptune@2.19.0", when="+hypre", type=("build", "run"))
+
+    depends_on("openmpi@4:", when="+mpispawn", type=("build", "run"))
+    conflicts("mpich", when="+mpispawn")
+    conflicts("spectrum-mpi", when="+mpispawn")
+    conflicts("cray-mpich", when="+mpispawn")
+    conflicts("gcc@:7")
 
     def cmake_args(self):
         spec = self.spec
         fc_flags = []
-        if '%gcc@10:' in spec or self.spec.satisfies('%apple-clang@11:'):
-            fc_flags.append('-fallow-argument-mismatch')
+        if "%gcc@10:" in spec or self.spec.satisfies("%apple-clang@11:"):
+            fc_flags.append("-fallow-argument-mismatch")
 
         args = [
-            '-DGPTUNE_INSTALL_PATH=%s' % python_platlib,
-            '-DTPL_BLAS_LIBRARIES=%s' % spec['blas'].libs.joined(";"),
-            '-DTPL_LAPACK_LIBRARIES=%s' % spec['lapack'].libs.joined(";"),
-            '-DTPL_SCALAPACK_LIBRARIES=%s' % spec['scalapack'].
-            libs.joined(";"),
-            '-DCMAKE_Fortran_FLAGS=' + ''.join(fc_flags),
-            '-DBUILD_SHARED_LIBS=ON',
-            '-DCMAKE_CXX_COMPILER=%s' % spec['mpi'].mpicxx,
-            '-DCMAKE_C_COMPILER=%s' % spec['mpi'].mpicc,
-            '-DCMAKE_Fortran_COMPILER=%s' % spec['mpi'].mpifc,
+            "-DGPTUNE_INSTALL_PATH=%s" % python_platlib,
+            "-DTPL_BLAS_LIBRARIES=%s" % spec["blas"].libs.joined(";"),
+            "-DTPL_LAPACK_LIBRARIES=%s" % spec["lapack"].libs.joined(";"),
+            "-DTPL_SCALAPACK_LIBRARIES=%s" % spec["scalapack"].libs.joined(";"),
+            "-DCMAKE_Fortran_FLAGS=" + "".join(fc_flags),
+            "-DBUILD_SHARED_LIBS=ON",
+            "-DCMAKE_CXX_COMPILER=%s" % spec["mpi"].mpicxx,
+            "-DCMAKE_C_COMPILER=%s" % spec["mpi"].mpicc,
+            "-DCMAKE_Fortran_COMPILER=%s" % spec["mpi"].mpifc,
         ]
 
         return args
 
-    examples_src_dir = 'examples'
-    src_dir = 'GPTune'
+    examples_src_dir = "examples"
+    src_dir = "GPTune"
     nodes = 1
     cores = 4
 
-    @run_after('install')
+    @run_after("install")
     def cache_test_sources(self):
         """Copy the example source files after the package is installed to an
         install test subdirectory for use during `spack test run`."""
         self.cache_extra_test_sources([self.examples_src_dir])
 
     def setup_run_environment(self, env):
-        env.set('GPTUNE_INSTALL_PATH', python_platlib)
+        env.set("GPTUNE_INSTALL_PATH", python_platlib)
 
     def test(self):
         spec = self.spec
         comp_name = self.compiler.name
-        comp_version = str(self.compiler.version).replace('.', ',')
-        test_dir = join_path(self.test_suite.current_test_cache_dir,
-                             self.examples_src_dir)
-
-        if '+superlu' in spec:
-            superludriver = join_path(spec['superlu-dist'].prefix.lib,
-                                      'EXAMPLE/pddrive_spawn')
-            op = ['-r', superludriver, '.']
+        comp_version = str(self.compiler.version).replace(".", ",")
+        test_dir = join_path(self.test_suite.current_test_cache_dir, self.examples_src_dir)
+
+        if "+superlu" in spec:
+            superludriver = join_path(spec["superlu-dist"].prefix.lib, "EXAMPLE/pddrive_spawn")
+            op = ["-r", superludriver, "."]
             # copy superlu-dist executables to the correct place
-            wd = join_path(test_dir, 'SuperLU_DIST')
-            self.run_test('rm', options=['-rf', 'superlu_dist'], work_dir=wd)
-            self.run_test('git', options=['clone', 'https://github.com/xiaoyeli/superlu_dist.git'], work_dir=wd)
-            self.run_test('mkdir', options=['-p',
-                                            'build'], work_dir=wd + '/superlu_dist')
-            self.run_test('mkdir', options=['-p', 'EXAMPLE'],
-                          work_dir=wd + '/superlu_dist/build')
-            self.run_test('cp', options=op, work_dir=wd + '/superlu_dist/build/EXAMPLE')
-
-        if '+hypre' in spec:
-            hypredriver = join_path(spec['hypre'].prefix.bin, 'ij')
-            op = ['-r', hypredriver, '.']
+            wd = join_path(test_dir, "SuperLU_DIST")
+            self.run_test("rm", options=["-rf", "superlu_dist"], work_dir=wd)
+            self.run_test(
+                "git",
+                options=["clone", "https://github.com/xiaoyeli/superlu_dist.git"],
+                work_dir=wd,
+            )
+            self.run_test("mkdir", options=["-p", "build"], work_dir=wd + "/superlu_dist")
+            self.run_test("mkdir", options=["-p", "EXAMPLE"], work_dir=wd + "/superlu_dist/build")
+            self.run_test("cp", options=op, work_dir=wd + "/superlu_dist/build/EXAMPLE")
+
+        if "+hypre" in spec:
+            hypredriver = join_path(spec["hypre"].prefix.bin, "ij")
+            op = ["-r", hypredriver, "."]
             # copy superlu-dist executables to the correct place
-            wd = join_path(test_dir, 'Hypre')
-            self.run_test('rm', options=['-rf', 'hypre'], work_dir=wd)
-            self.run_test('git', options=['clone', 'https://github.com/hypre-space/hypre.git'], work_dir=wd)
-            self.run_test('cp', options=op, work_dir=wd + '/hypre/src/test/')
+            wd = join_path(test_dir, "Hypre")
+            self.run_test("rm", options=["-rf", "hypre"], work_dir=wd)
+            self.run_test(
+                "git", options=["clone", "https://github.com/hypre-space/hypre.git"], work_dir=wd
+            )
+            self.run_test("cp", options=op, work_dir=wd + "/hypre/src/test/")
 
         wd = self.test_suite.current_test_cache_dir
-        with open('{0}/run_env.sh'.format(wd), 'w') as envfile:
+        with open("{0}/run_env.sh".format(wd), "w") as envfile:
             envfile.write('if [[ $NERSC_HOST = "cori" ]]; then\n')
-            envfile.write('    export machine=cori\n')
+            envfile.write("    export machine=cori\n")
             envfile.write('elif [[ $(uname -s) = "Darwin" ]]; then\n')
-            envfile.write('    export machine=mac\n')
-            envfile.write('elif [[ $(dnsdomainname) = ' +
-                          '"summit.olcf.ornl.gov" ]]; then\n')
-            envfile.write('    export machine=summit\n')
-            envfile.write('elif [[ $(cat /etc/os-release | grep "PRETTY_NAME") ==' +
-                          ' *"Ubuntu"* || $(cat /etc/os-release | grep' +
-                          ' "PRETTY_NAME") == *"Debian"* ]]; then\n')
-            envfile.write('    export machine=unknownlinux\n')
-            envfile.write('fi\n')
-            envfile.write('export GPTUNEROOT=$PWD\n')
-            envfile.write('export MPIRUN={0}\n'.format
-                          (which(spec['mpi'].prefix.bin + '/mpirun')))
-            envfile.write('export PYTHONPATH={0}:$PYTHONPATH\n'.format
-                          (python_platlib + '/gptune'))
-            envfile.write('export proc=$(spack arch)\n')
-            envfile.write('export mpi={0}\n'.format(spec['mpi'].name))
-            envfile.write('export compiler={0}\n'.format(comp_name))
-            envfile.write('export nodes={0} \n'.format(self.nodes))
-            envfile.write('export cores={0} \n'.format(self.cores))
-            envfile.write('export ModuleEnv=$machine-$proc-$mpi-$compiler \n')
-            envfile.write('software_json=$(echo ",\\\"software_configuration\\\":' +
-                          '{\\\"' + spec['blas'].name +
-                          '\\\":{\\\"version_split\\\":' +
-                          ' [' + str(spec['blas'].versions).replace('.', ',') +
-                          ']},\\\"' + spec['mpi'].name +
-                          '\\\":{\\\"version_split\\\": [' +
-                          str(spec['mpi'].versions).replace('.', ',') + ']},\\\"' +
-                          spec['scalapack'].name +
-                          '\\\":{\\\"version_split\\\": [' +
-                          str(spec['scalapack'].versions).replace('.', ',') +
-                          ']},\\\"' +
-                          str(comp_name) + '\\\":{\\\"version_split\\\": [' +
-                          str(comp_version) + ']}}") \n')
-            envfile.write('loadable_software_json=$(echo ",\\\"loadable_software_' +
-                          'configurations\\\":{\\\"' + spec['blas'].name +
-                          '\\\":{\\\"version_split\\\": [' +
-                          str(spec['blas'].versions).replace('.', ',') +
-                          ']},\\\"' + spec['mpi'].name +
-                          '\\\":{\\\"version_split\\\": [' +
-                          str(spec['mpi'].versions).replace('.', ',') + ']},\\\"' +
-                          spec['scalapack'].name +
-                          '\\\":{\\\"version_split\\\": [' +
-                          str(spec['scalapack'].versions).replace('.', ',') +
-                          ']},\\\"' + str(comp_name) +
-                          '\\\":{\\\"version_split\\\": ['
-                          + str(comp_version) + ']}}") \n')
-            envfile.write('machine_json=$(echo ",\\\"machine_configuration\\\":' +
-                          '{\\\"machine_name\\\":\\\"$machine\\\",\\\"$proc\\\":' +
-                          '{\\\"nodes\\\":$nodes,\\\"cores\\\":$cores}}") \n')
-            envfile.write('loadable_machine_json=$(echo ",\\\"loadable_machine_' +
-                          'configurations\\\":{\\\"$machine\\\":{\\\"$proc\\\":' +
-                          '{\\\"nodes\\\":$nodes,\\\"cores\\\":$cores}}}") \n')
+            envfile.write("    export machine=mac\n")
+            envfile.write("elif [[ $(dnsdomainname) = " + '"summit.olcf.ornl.gov" ]]; then\n')
+            envfile.write("    export machine=summit\n")
+            envfile.write(
+                'elif [[ $(cat /etc/os-release | grep "PRETTY_NAME") =='
+                + ' *"Ubuntu"* || $(cat /etc/os-release | grep'
+                + ' "PRETTY_NAME") == *"Debian"* ]]; then\n'
+            )
+            envfile.write("    export machine=unknownlinux\n")
+            envfile.write("fi\n")
+            envfile.write("export GPTUNEROOT=$PWD\n")
+            envfile.write("export MPIRUN={0}\n".format(which(spec["mpi"].prefix.bin + "/mpirun")))
+            envfile.write("export PYTHONPATH={0}:$PYTHONPATH\n".format(python_platlib + "/gptune"))
+            envfile.write("export proc=$(spack arch)\n")
+            envfile.write("export mpi={0}\n".format(spec["mpi"].name))
+            envfile.write("export compiler={0}\n".format(comp_name))
+            envfile.write("export nodes={0} \n".format(self.nodes))
+            envfile.write("export cores={0} \n".format(self.cores))
+            envfile.write("export ModuleEnv=$machine-$proc-$mpi-$compiler \n")
+            envfile.write(
+                'software_json=$(echo ",\\"software_configuration\\":'
+                + '{\\"'
+                + spec["blas"].name
+                + '\\":{\\"version_split\\":'
+                + " ["
+                + str(spec["blas"].versions).replace(".", ",")
+                + ']},\\"'
+                + spec["mpi"].name
+                + '\\":{\\"version_split\\": ['
+                + str(spec["mpi"].versions).replace(".", ",")
+                + ']},\\"'
+                + spec["scalapack"].name
+                + '\\":{\\"version_split\\": ['
+                + str(spec["scalapack"].versions).replace(".", ",")
+                + ']},\\"'
+                + str(comp_name)
+                + '\\":{\\"version_split\\": ['
+                + str(comp_version)
+                + ']}}") \n'
+            )
+            envfile.write(
+                'loadable_software_json=$(echo ",\\"loadable_software_'
+                + 'configurations\\":{\\"'
+                + spec["blas"].name
+                + '\\":{\\"version_split\\": ['
+                + str(spec["blas"].versions).replace(".", ",")
+                + ']},\\"'
+                + spec["mpi"].name
+                + '\\":{\\"version_split\\": ['
+                + str(spec["mpi"].versions).replace(".", ",")
+                + ']},\\"'
+                + spec["scalapack"].name
+                + '\\":{\\"version_split\\": ['
+                + str(spec["scalapack"].versions).replace(".", ",")
+                + ']},\\"'
+                + str(comp_name)
+                + '\\":{\\"version_split\\": ['
+                + str(comp_version)
+                + ']}}") \n'
+            )
+            envfile.write(
+                'machine_json=$(echo ",\\"machine_configuration\\":'
+                + '{\\"machine_name\\":\\"$machine\\",\\"$proc\\":'
+                + '{\\"nodes\\":$nodes,\\"cores\\":$cores}}") \n'
+            )
+            envfile.write(
+                'loadable_machine_json=$(echo ",\\"loadable_machine_'
+                + 'configurations\\":{\\"$machine\\":{\\"$proc\\":'
+                + '{\\"nodes\\":$nodes,\\"cores\\":$cores}}}") \n'
+            )
 
         # copy the environment configuration files to non-cache directories
-        op = ['run_env.sh', python_platlib + '/gptune/.']
-        self.run_test('cp', options=op, work_dir=wd)
-        op = ['run_env.sh', self.install_test_root + '/.']
-        self.run_test('cp', options=op, work_dir=wd)
-
-        apps = ['Scalapack-PDGEQRF_RCI']
-        if '+mpispawn' in spec:
-            apps = apps + ['GPTune-Demo', 'Scalapack-PDGEQRF']
-        if '+superlu' in spec:
-            apps = apps + ['SuperLU_DIST_RCI']
-            if '+mpispawn' in spec:
-                apps = apps + ['SuperLU_DIST']
-        if '+hypre' in spec:
-            if '+mpispawn' in spec:
-                apps = apps + ['Hypre']
+        op = ["run_env.sh", python_platlib + "/gptune/."]
+        self.run_test("cp", options=op, work_dir=wd)
+        op = ["run_env.sh", self.install_test_root + "/."]
+        self.run_test("cp", options=op, work_dir=wd)
+
+        apps = ["Scalapack-PDGEQRF_RCI"]
+        if "+mpispawn" in spec:
+            apps = apps + ["GPTune-Demo", "Scalapack-PDGEQRF"]
+        if "+superlu" in spec:
+            apps = apps + ["SuperLU_DIST_RCI"]
+            if "+mpispawn" in spec:
+                apps = apps + ["SuperLU_DIST"]
+        if "+hypre" in spec:
+            if "+mpispawn" in spec:
+                apps = apps + ["Hypre"]
 
         for app in apps:
             wd = join_path(test_dir, app)
-            self.run_test('bash', options=['run_examples.sh'], work_dir=wd,
-                          purpose='gptune smoke test for {0}'.format(app))
+            self.run_test(
+                "bash",
+                options=["run_examples.sh"],
+                work_dir=wd,
+                purpose="gptune smoke test for {0}".format(app),
+            )