feat(python): warn on fork (#2258)

Closes #768
This commit is contained in:
Will Jones
2025-03-21 17:18:10 -07:00
committed by GitHub
parent 93a82fd371
commit abe06fee3d
2 changed files with 15 additions and 0 deletions

View File

@@ -7,6 +7,7 @@ import os
from concurrent.futures import ThreadPoolExecutor
from datetime import timedelta
from typing import Dict, Optional, Union, Any
import warnings
__version__ = importlib.metadata.version("lancedb")
@@ -213,3 +214,13 @@ __all__ = [
"RemoteDBConnection",
"__version__",
]
def __warn_on_fork():
warnings.warn(
"lance is not fork-safe. If you are using multiprocessing, use spawn instead.",
)
if hasattr(os, "register_at_fork"):
os.register_at_fork(before=__warn_on_fork)