apply rounding to similarity values

YerevaNN · Apr 2, 2024 · 7c12ab9 · 7c12ab9
1 parent 10208db
commit 7c12ab9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/chemlactica/utils/text_format_utils.py b/chemlactica/utils/text_format_utils.py
@@ -111,7 +111,8 @@ def format_key_value(key, value, rng):
             # value = random.sample(value, 5)
             value = rng.choice(value, size=10, replace=False, shuffle=False)
         for pair in value:
-            formatted_string += f"{SPECIAL_TAGS['similarity']['start']}{pair['SMILES']} {pair['similarity']}{SPECIAL_TAGS['similarity']['end']}"  # noqa
+            rounded_sim = "{:.2f}".format(float(pair["similarity"]))
+            formatted_string += f"{SPECIAL_TAGS['similarity']['start']}{pair['SMILES']} {rounded_sim}{SPECIAL_TAGS['similarity']['end']}"  # noqa
     elif key == "experimental":
         for pair in value:
             formatted_string += (