Running into bug when using knn graph in Datalab Tabular tutorial

[mturk24]

Running into the following error:

Error in non_iid: If a knn_graph is not provided, features must be provided to fit a new knn.
Failed to check for these issue types: [NonIIDIssueManager]

Screenshot shows error and this is the exact code being run from the tutorial

KNN = NearestNeighbors(metric='euclidean')
KNN.fit(X_processed.values)

knn_graph = KNN.kneighbors_graph(mode="distance")

data = {"X": X_processed.values, "y": labels}

lab = Datalab(data, label_name="y")
lab.find_issues(pred_probs=pred_probs, knn_graph=knn_graph)

[coding-famer]

I'd like to work on this issue but I can't reproduce it. The tutorial works well on my machine and on the document website. And the error message for non_iid is

"If a knn_graph is not provided, either 'features' or 'pred_probs' must be provided to fit a new knn."

instead of features must be provided to fit a new knn.

Can you provide more details about reproducing this error?