Skip to content

Fix failing unit tests for pandas and mlflow compatibility #1463

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
skrawcz merged 10 commits into from
Feb 5, 2026
Merged

Fix failing unit tests for pandas and mlflow compatibility #1463

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
f141d7a
Enhance visualization with display_name tag support
Exudev Jan 17, 2026
9eeb33e
Updates the assertion in the graph visualization test to ensure that …
Exudev Jan 17, 2026
4de0fa6
adjusting the formatting of the header.
Exudev Jan 17, 2026
8541627
Enhance display_name handling in graph visualizations
Exudev Jan 17, 2026
edf11e9
Refactor display_name imports in graph visualization tests
Exudev Jan 20, 2026
cbc14c8
Update visualization documentation to clarify usage of display_name d…
Exudev Jan 20, 2026
41537a6
Refactor display_name imports in test_graph.py to streamline test dep…
Exudev Jan 24, 2026
e80c28d
Merge branch 'apache:main' into main
Exudev Jan 27, 2026
a342a33
Fix unit tests for external library changes
Exudev Feb 2, 2026
635405e
Merge branch 'apache:main' into main
Exudev Feb 2, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion tests/plugins/test_mlflow_extension.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,9 +17,12 @@

from pathlib import Path

import mlflow
import numpy as np
import pytest

# Skip entire module if mlflow is not available (e.g., Python 3.14+)
mlflow = pytest.importorskip("mlflow")

from sklearn.base import BaseEstimator
from sklearn.linear_model import LinearRegression

Expand Down
8 changes: 5 additions & 3 deletions tests/test_base.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -481,17 +481,19 @@ def test_StrictIndexTypePandasDataFrameResult_build_result(outputs, expected_res
),
"series2": pd.Series(
[4, 5, 6],
index=pd.PeriodIndex(year=[2022, 2022, 2022], month=[1, 2, 3], freq="M"),
index=pd.PeriodIndex.from_fields(
year=[2022, 2022, 2022], month=[1, 2, 3], freq="M"
),
),
"series3": pd.Series(
[4, 5, 6],
index=pd.PeriodIndex(
index=pd.PeriodIndex.from_fields(
year=[2022, 2022, 2022], month=[1, 1, 1], day=[3, 4, 5], freq="B"
),
),
"series4": pd.Series(
[4, 5, 6],
index=pd.PeriodIndex(
index=pd.PeriodIndex.from_fields(
year=[2022, 2022, 2022], month=[1, 1, 1], day=[4, 11, 18], freq="W"
),
),
Expand Down
Loading