diff --git a/python/paddle/io/dataloader/worker.py b/python/paddle/io/dataloader/worker.py
index a559a616bb2963..b1284a646f656c 100644
--- a/python/paddle/io/dataloader/worker.py
+++ b/python/paddle/io/dataloader/worker.py
@@ -11,6 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+from __future__ import annotations
 
 import os
 import queue
@@ -76,7 +77,7 @@ def is_alive(self):
 _worker_info = None
 
 
-def get_worker_info():
+def get_worker_info() -> WorkerInfo | None:
     """
     Get DataLoader worker process information function, this function is
     used to split data copy in worker process for IterableDataset
@@ -117,8 +118,8 @@ def get_worker_info():
             ...         else:
             ...             per_worker = int(
             ...                 math.ceil((self.end - self.start) / float(
-            ...                     worker_info.num_workers)))
-            ...             worker_id = worker_info.id
+            ...                     worker_info.num_workers)))  # type: ignore[attr-defined]
+            ...             worker_id = worker_info.id  # type: ignore[attr-defined]
             ...             iter_start = self.start + worker_id * per_worker
             ...             iter_end = min(iter_start + per_worker, self.end)
             ...