Update loss_functions.py

Author: Addyk-24

machine_learning/loss_functions.py

@@ -684,14 +684,9 @@ def root_mean_squared_error(y_true: np.array, y_pred: np.array) -> float:
 
     >>> true_labels = np.array([100, 200, 300])
     >>> predicted_probs = np.array([110, 190, 310])
-    >>> round(root_mean_squared_error(true_labels, predicted_probs), 4)
+    >>> float(root_mean_squared_error(true_labels, predicted_probs))
     10.0
 
-    >>> true_labels = [2, 4, 6, 8]
-    >>> predicted_probs = [3, 5, 7, 10]
-    >>> round(root_mean_squared_error(true_labels, predicted_probs), 4)
-    1.3228756555322954
-
     >>> true_labels = np.array([1.0, 2.0, 3.0, 4.0, 5.0])
     >>> predicted_probs = np.array([0.3, 0.8, 0.9, 0.2])
     >>> root_mean_squared_error(true_labels, predicted_probs)