We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
tkinter
1 parent f8f18f9 commit abc4562Copy full SHA for abc4562
tests/test_models/conftest.py
@@ -1,10 +1,19 @@
1
-import numpy as np
+import os
2
+
3
import pytest
4
+from skbase.utils.dependencies import _check_soft_dependencies
5
import torch
6
7
from pytorch_forecasting import TimeSeriesDataSet
8
from pytorch_forecasting.data import EncoderNormalizer, GroupNormalizer, NaNLabelEncoder
-from pytorch_forecasting.data.examples import generate_ar_data, get_stallion_data
9
10
+# used to prevent tkinter related errors in CI, see #1937
11
+if _check_soft_dependencies("matplotlib", severity="none"):
12
+ if os.environ.get("GITHUB_ACTIONS") == "true":
13
+ import matplotlib
14
15
+ matplotlib.use("Agg")
16
17
18
torch.manual_seed(23)
19
0 commit comments