mark as unstable

pola-rs · May 19, 2024 · 6eef9b9 · 6eef9b9
1 parent d9f1386
commit 6eef9b9
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/py-polars/polars/dataframe/frame.py b/py-polars/polars/dataframe/frame.py
@@ -1635,6 +1635,7 @@ def to_jax(
         order: IndexOrder = ...,
     ) -> dict[str, jax.Array]: ...
 
+    @unstable()
     def to_jax(
         self,
         return_type: JaxExportType = "array",
@@ -1650,6 +1651,10 @@ def to_jax(
 
         .. versionadded:: 0.20.27
 
+        .. warning::
+            This functionality is currently considered **unstable**. It may be
+            changed at any point without it being considered a breaking change.
+
         Parameters
         ----------
         return_type : {"array", "dict"}
@@ -1838,6 +1843,7 @@ def to_torch(
         dtype: PolarsDataType | None = ...,
     ) -> dict[str, torch.Tensor]: ...
 
+    @unstable()
     def to_torch(
         self,
         return_type: TorchExportType = "tensor",
@@ -1851,6 +1857,10 @@ def to_torch(
 
         .. versionadded:: 0.20.23
 
+        .. warning::
+            This functionality is currently considered **unstable**. It may be
+            changed at any point without it being considered a breaking change.
+
         Parameters
         ----------
         return_type : {"tensor", "dataset", "dict"}