ipywidgets in the enviornment on vscode jupyter notebook cause cell with pymc.sample to hang with default sampler

[danieltomasz]

Describe the issue:

in the environments without ipywidgets the inference taking 1 second
When the ipywidgets is installed the cell with pymc code hangs on

Initializing NUTS using jitter+adapt_diag...
Multiprocess sampling (4 chains in 4 jobs)
NUTS: [θ]

I tested install both via conda and pip, the inference via bambi and other samplers worked fine in the same env,
the example below I tested in clean envs with

matplotlib
numpy
pandas
pymc
preliz
ipykernel
arviz

installed (and with or without ipywidgets)

Reproduceable code example:

import numpy as np
import pandas as pd
import pymc as pm
import preliz as pz
np.random.seed(123)
trials = 4
theta_real = 0.35 # unknown value in a real experiment
data = pz.Binomial(n=1, p=theta_real).rvs(trials)
data

with pm.Model() as our_first_model:
    θ = pm.Beta('θ', alpha=1., beta=1.)
    y = pm.Bernoulli('y', p=θ, observed=data)
    idata = pm.sample(1000, random_seed=4591)

Error message:

No response

PyMC version information:

Python 3.12.8, Pymc 5.19.1

Context for the issue:

Version: 1.96.0
Commit: 138f619c86f1199955d53b4166bef66ef252935c
Date: 2024-12-11T02:29:09.626Z
Electron: 32.2.6
ElectronBuildId: 10629634
Chromium: 128.0.6613.186
Node.js: 20.18.1
V8: 12.8.374.38-electron.0
OS: Darwin arm64 24.2.0

[welcome]

]
🎉 Welcome to PyMC! 🎉 We're really excited to have your input into the project! 💖

If you haven't done so already, please make sure you check out our Contributing Guidelines and Code of Conduct.

[danieltomasz]

This seems to be happening only in vscode stable, vscode insiders and jupyter notebooks/lab seems to be fine; maybe this is issue with my environments and plugins in vscode, would be great if someone can reproduce and confirm;
Downgrading ipywidgets and pymc doesn't help

[aloctavodia]

I am not able to reproduce . I tested in two different environments.

vscode version 1.94
python 3.11.8
PyMC 5.18.2 or 5.19.1
ipywidgets 8.1.1 or 8.0.4

[danieltomasz]

thanks for your time and for checking @aloctavodia, I will close this issue, since this is probably not pymc related, but vscode+ipywidgets+vscode plugins interactions