Skip to content

Commit 9172925

Browse files
ricardoV94twiecki
authored andcommitted
Remove experimental module warning from jax.py
1 parent 301b28a commit 9172925

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

pymc/sampling/jax.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@
1414
import os
1515
import re
1616
import sys
17-
import warnings
1817

1918
from functools import partial
2019
from typing import Any, Callable, Dict, List, Optional, Sequence, Union
@@ -49,9 +48,6 @@
4948
from pymc.logprob.utils import CheckParameterValue
5049
from pymc.util import RandomSeed, _get_seeds_per_chain, get_default_varnames
5150

52-
warnings.warn("This module is experimental.")
53-
54-
5551
__all__ = (
5652
"get_jaxified_graph",
5753
"get_jaxified_logp",

0 commit comments

Comments
 (0)