A code snippet is hung forever in edition2-alibi-detect-examples in drift/Chapter03

[bit-scientist]

The below code cell is running forever:

n_runs = 4
#times_h0 = [time_run(cd, X_h0, window_size) for _ in range(n_runs)]
times_h0 = [time_run(cd, X_ref, window_size) for _ in range(n_runs)] #changed to X_ref just to run a test, change back!


print(f"Average run-time under no-drift: {np.mean(times_h0)}")
_ = scipy.stats.probplot(np.array(times_h0), dist=scipy.stats.geom, sparams=1/ert, plot=plt)

I am not sure how to tackle this. Running on Ubuntu 24.04.2 LTS (WLS)