format

Author: lwshanbd

llvm_ir_dataset_utils/builders/portage_builder.py

@@ -17,7 +17,6 @@
 from llvm_ir_dataset_utils.util import portage as portage_utils
 from llvm_ir_dataset_utils.util import extract_source_lib
 
-
 BUILD_LOG_NAME = './portage_build.log'
 
 
@@ -73,7 +72,7 @@ def extract_ir(package_spec, corpus_dir, build_dir, threads):
   build_directory = build_dir + "/portage/"
   package_spec = package_spec + "*"
   match = glob.glob(os.path.join(build_directory, package_spec))
-  assert(len(match) == 1)
+  assert (len(match) == 1)
   package_name_with_version = os.path.basename(match[0])
   build_directory = match[0] + "/work/" + package_name_with_version
   if build_directory is not None:

llvm_ir_dataset_utils/util/portage.py

@@ -6,23 +6,20 @@
 
 
 def get_portage_compiler_config(filename):
-    content = (
-        'COMMON_FLAGS="-O2 -pipe -Xclang -fembed-bitcode=all"\n'
-        '\n'
-        'CC="/root/ir-dataset/utils/compiler_wrapper"\n'
-        'CXX="/root/ir-dataset/utils/compiler_wrapper++"\n'
-        'CFLAGS="${COMMON_FLAGS}"\n'
-        'CXXFLAGS="${COMMON_FLAGS}"\n'
-        'FCFLAGS="${COMMON_FLAGS}"\n'
-        'FFLAGS="${COMMON_FLAGS}"\n'
-        '\n'
-        'FEATURES="noclean"\n'
-        '\n'
-        'LC_MESSAGES=C.utf8'
-    )
-    with open(filename, 'w') as file:
-        file.write(content)
-
+  content = ('COMMON_FLAGS="-O2 -pipe -Xclang -fembed-bitcode=all"\n'
+             '\n'
+             'CC="/root/ir-dataset/utils/compiler_wrapper"\n'
+             'CXX="/root/ir-dataset/utils/compiler_wrapper++"\n'
+             'CFLAGS="${COMMON_FLAGS}"\n'
+             'CXXFLAGS="${COMMON_FLAGS}"\n'
+             'FCFLAGS="${COMMON_FLAGS}"\n'
+             'FFLAGS="${COMMON_FLAGS}"\n'
+             '\n'
+             'FEATURES="noclean"\n'
+             '\n'
+             'LC_MESSAGES=C.utf8')
+  with open(filename, 'w') as file:
+    file.write(content)
 
 
 def portage_setup_compiler(build_dir):
@@ -41,8 +38,9 @@ def portage_setup_compiler(build_dir):
   os.symlink('/etc/portage/make.profile', make_profile_path)
   get_portage_compiler_config(make_conf_path)
 
+
 def clean_binpkg(package_spec):
   command_vector = ['rm', '-rf', '/var/cache/binpkgs/' + package_spec]
   subprocess.run(command_vector)
   sync_command = ['emaint', '--fix', 'binhost']
-  subprocess.run(sync_command)
+  subprocess.run(sync_command)