Hi,

I am running some code which fits a linear model via Statsmodel. Everytime I run the .fit function it generates the error: "Error displaying widget: undefined".

I can reproduce the error via the simple code below.

I am not sure what is causing this issue and I cannot find any similar issues online. Is anybody aware of the issue causing this error?

Node type: Standard_DS3_v2

Databricks runtime version: 15.1 ML

Statsmodel version: 0.14.0

import statsmodels.api as sm

import pandas as pd

# Create sample data

data = {

    "X": [1, 2, 3, 4, 5],

    "Y": [2, 4, 5, 4, 5]

}

df = pd.DataFrame(data)

# Define independent and dependent variables

X = sm.add_constant(df["X"])  # Add constant for intercept

y = df["Y"]

# Fit OLS model

model = sm.OLS(y, X).fit()