missed one more instance of idxmax

Author: vsoch

singularity/build/main.py

@@ -162,6 +162,7 @@ def run_build(build_dir,params,verbose=True, compress_image=False):
         diff = get_diff(image_package=image_package)
 
         # Inspect to get labels and other metadata
+
         cli = Singularity(debug=params['debug'])
         inspect = cli.inspect(image_path=image)
 
@@ -179,7 +180,8 @@ def run_build(build_dir,params,verbose=True, compress_image=False):
         extensions = extension_counts(diff=diff)
 
         os_sims = estimate_os(image_package=image_package,return_top=False)
-        most_similar = os_sims['SCORE'].idxmax()
+        most_similar = os_sims['SCORE'].values.argmax()
+        most_similar = os_sims['SCORE'].index.tolist()[most_similar]
 
         metrics = {'build_time_seconds':final_time,
                    'singularity_version':singularity_version,