diff --git a/machine_learning/ridge_regression.py b/machine_learning/ridge_regression.py
index 20324e987..32c76a90c 100644
--- a/machine_learning/ridge_regression.py
+++ b/machine_learning/ridge_regression.py
@@ -1,7 +1,5 @@
 import numpy as np
 import pandas as pd
-from typing import Optional, Tuple
-
 
 class RidgeRegression:
     def __init__(
@@ -16,9 +14,7 @@ class RidgeRegression:
         self.alpha = alpha
         self.lambda_ = lambda_
         self.iterations = iterations
-        self.theta: Optional[np.ndarray] = (
-            None  # Initialize as None, later will be ndarray
-        )
+        self.theta: np.ndarray | None = None  # Initialize as None, later will be ndarray
 
     def feature_scaling(
         self, features: np.ndarray