Add support python 3.11

.github/workflows/publish-to-test-pypi.yml

@@ -11,10 +11,10 @@ jobs:
     runs-on: ubuntu-latest
     steps:
     - uses: actions/checkout@master
-    - name: Set up Python 3.10
+    - name: Set up Python 3.11
       uses: actions/setup-python@v3
       with:
-        python-version: "3.10"
+        python-version: "3.11"
 
     - name: Install pypa/build
       run: >-

.github/workflows/test.yml

@@ -11,7 +11,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.8", "3.9", "3.10"]
+        python-version: ["3.8", "3.9", "3.10", "3.11"]
 
     name: Set up Python ${{ matrix.python-version }}
     steps:

.pylintrc

@@ -63,9 +63,7 @@ disable=missing-docstring,
         too-many-branches,
         too-many-statements,
         too-many-instance-attributes,
-        no-self-use,
         too-few-public-methods,
-        bad-continuation,
         import-error,
         protected-access
 
@@ -136,13 +134,6 @@ max-line-length=100
 # Maximum number of lines in a module
 max-module-lines=1000
 
-# List of optional constructs for which whitespace checking is disabled. `dict-
-# separator` is used to allow tabulation in dicts, etc.: {1  : 1,\n222: 2}.
-# `trailing-comma` allows a space between comma and closing bracket: (a, ).
-# `empty-line` allows space-only lines.
-no-space-check=trailing-comma,
-               dict-separator
-
 # Allow the body of a class to be on the same line as the declaration if body
 # contains single statement.
 single-line-class-stmt=no

pymc_bart/bart.py

@@ -14,9 +14,9 @@
 #   See the License for the specific language governing permissions and
 #   limitations under the License.
 
+import warnings
 from multiprocessing import Manager
 from typing import List, Optional, Tuple
-import warnings
 
 import numpy as np
 import numpy.typing as npt
@@ -26,9 +26,9 @@
 from pymc.logprob.abstract import _logprob
 from pytensor.tensor.random.op import RandomVariable
 
+from .split_rules import SplitRule
 from .tree import Tree
 from .utils import TensorLike, _sample_posterior
-from .split_rules import SplitRule
 
 __all__ = ["BART"]
 
@@ -47,7 +47,7 @@ def _supp_shape_from_params(self, dist_params, rep_param_idx=1, param_shapes=Non
         return dist_params[0].shape[:1]
 
     @classmethod
-    def rng_fn(
+    def rng_fn(  # pylint: disable=W0237
         cls, rng=None, X=None, Y=None, m=None, alpha=None, beta=None, split_prior=None, size=None
     ):
         if not cls.all_trees:
@@ -147,21 +147,21 @@ def __new__(
         bart_op = type(
             f"BART_{name}",
             (BARTRV,),
-            dict(
-                name="BART",
-                all_trees=cls.all_trees,
-                inplace=False,
-                initval=Y.mean(),
-                X=X,
-                Y=Y,
-                m=m,
-                response=response,
-                alpha=alpha,
-                beta=beta,
-                split_prior=split_prior,
-                split_rules=split_rules,
-                separate_trees=separate_trees,
-            ),
+            {
+                "name": "BART",
+                "all_trees": cls.all_trees,
+                "inplace": False,
+                "initval": Y.mean(),
+                "X": X,
+                "Y": Y,
+                "m": m,
+                "response": response,
+                "alpha": alpha,
+                "beta": beta,
+                "split_prior": split_prior,
+                "split_rules": split_rules,
+                "separate_trees": separate_trees,
+            },
         )()
 
         Distribution.register(BARTRV)

pymc_bart/utils.py

@@ -246,7 +246,11 @@ def plot_ice(
     rng = np.random.default_rng(random_seed)
 
     if func is None:
-        func = lambda x: x
+
+        def identity(x):
+            return x
+
+        func = identity
 
     (
         X,
@@ -392,7 +396,11 @@ def plot_pdp(
     rng = np.random.default_rng(random_seed)
 
     if func is None:
-        func = lambda x: x
+
+        def identity(x):
+            return x
+
+        func = identity
 
     (
         X,

requirements-dev.txt

@@ -3,6 +3,6 @@ click==8.0.4
 mypy>=1.1.1
 pandas-stubs==1.5.3.230304
 pre-commit
-pylint==2.10.2
+pylint==2.17.4
 pytest-cov>=2.6.1
 pytest>=4.4.0

setup.py

@@ -32,6 +32,7 @@
     "Programming Language :: Python :: 3.8",
     "Programming Language :: Python :: 3.9",
     "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
     "License :: OSI Approved :: Apache Software License",
     "Intended Audience :: Science/Research",
     "Topic :: Scientific/Engineering",