diff --git a/flaml/automl/automl.py b/flaml/automl/automl.py
index 745f32971..1b6bef199 100644
--- a/flaml/automl/automl.py
+++ b/flaml/automl/automl.py
@@ -2,7 +2,6 @@
 #  * Copyright (c) FLAML authors. All rights reserved.
 #  * Licensed under the MIT License. See LICENSE file in the
 #  * project root for license information.
-from __future__ import annotations
 import time
 import os
 import sys
@@ -344,10 +343,7 @@ class AutoMLState:
         return sampled_X_train, sampled_y_train, sampled_weight, groups
 
     @staticmethod
-    def _compute_with_config_base(config_w_resource,
-                                  state: AutoMLState,
-                                  estimator,
-                                  is_report=True):
+    def _compute_with_config_base(config_w_resource, state, estimator, is_report=True):
         if "FLAML_sample_size" in config_w_resource:
             sample_size = int(config_w_resource["FLAML_sample_size"])
         else: