Fix behavioral change in extract_ir logging introduced in 846df06 (#41)

Author: mtrofin

compiler_opt/rl/inlining/config.py

@@ -98,6 +98,6 @@ def observation_processing_layer(obs_spec):
 
   return observation_processing_layer
 
+
 def get_nonnormalized_features():
   return ['reward', 'inlining_default', 'inlining_decision']
-

compiler_opt/rl/problem_configuration.py

@@ -100,5 +100,6 @@ def get_nonnormalized_features(self) -> Iterable[str]:
   def get_runner_type(self) -> 'type[compilation_runner.CompilationRunner]':
     raise NotImplementedError
 
+
 def is_thinlto(module_paths: Iterable[str]) -> bool:
   return tf.io.gfile.exists(next(iter(module_paths)) + '.thinlto.bc')

compiler_opt/rl/regalloc/config.py

@@ -98,8 +98,7 @@ def observation_processing_layer(obs_spec):
           first_non_zero = x
           break
 
-      normalize_fn = feature_ops.get_normalize_fn(quantile,
-                                                  with_sqrt,
+      normalize_fn = feature_ops.get_normalize_fn(quantile, with_sqrt,
                                                   with_z_score_normalization,
                                                   eps)
       log_normalize_fn = feature_ops.get_normalize_fn(
@@ -142,8 +141,9 @@ def progress_processing_fn(obs):
 
   return observation_processing_layer
 
+
 def get_nonnormalized_features():
-  return ['mask', 'nr_urgent',
-          'is_hint', 'is_local',
-          'is_free', 'max_stage',
-          'min_stage', 'reward']
+  return [
+      'mask', 'nr_urgent', 'is_hint', 'is_local', 'is_free', 'max_stage',
+      'min_stage', 'reward'
+  ]

compiler_opt/tools/extract_ir.py

@@ -271,7 +271,7 @@ def main(argv):
           f.write(path + '\n')
 
     logging.info('Converted %d files out of %d',
-                 len(objs) - len(relative_output_paths), len(objs))
+                 len(objs) - relative_output_paths.count(None), len(objs))
 
 
 if __name__ == '__main__':