diff --git a/py-polars/polars/catalog.py b/py-polars/polars/catalog.py
index 4ca70fdb55d2..504d95a9491b 100644
--- a/py-polars/polars/catalog.py
+++ b/py-polars/polars/catalog.py
@@ -5,7 +5,7 @@
 import os
 import sys
 from dataclasses import dataclass
-from typing import TYPE_CHECKING, Any, Generator, Literal
+from typing import TYPE_CHECKING, Any, Literal
 
 from polars._utils.unstable import issue_unstable_warning
 from polars._utils.wrap import wrap_ldf
@@ -13,6 +13,7 @@
 from polars.schema import Schema
 
 if TYPE_CHECKING:
+    from collections.abc import Generator
     from datetime import datetime
 
     from polars._typing import SchemaDict