use next() to return the formatted vector in axes map function

shaketune/graph_creators/axes_map_graph_creator.py

@@ -269,9 +269,8 @@ def _format_direction_vector(self, vectors: List[np.ndarray]) -> str:
                     axes_count[MACHINE_AXES[i]] += 1
                     break
 
-        # Check if all axes are present in the axes_map and return an error message if not
-        for _, count in axes_count.items():
-            if count != 1:
-                return 'unable to determine it correctly!'
-
-        return ', '.join(formatted_vector)
+        # If all axes are present, return the formatted vector
+        return next(
+            ('unable to determine it correctly!' for count in axes_count.values() if count != 1),
+            ', '.join(formatted_vector),
+        )