Skip to content

Commit 94e9fc8

Browse files
committed
fix tests
1 parent c55c3ab commit 94e9fc8

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

test/test_file_processor_pandas.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from __future__ import annotations
22

3-
import importlib
3+
import importlib.util
44
import tempfile
55

66
import pandas as pd

test/test_target.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import importlib.util
12
import io
23
import os
34
import shutil
@@ -8,13 +9,17 @@
89
import boto3
910
import numpy as np
1011
import pandas as pd
12+
import pytest
1113
from matplotlib import pyplot
1214
from moto import mock_aws
1315

1416
from gokart.file_processor import _ChunkedLargeFileReader
1517
from gokart.target import make_model_target, make_target
1618
from test.util import _get_temporary_directory
1719

20+
polars_installed = importlib.util.find_spec('polars') is not None
21+
pytestmark = pytest.mark.skipif(polars_installed, reason='polars installed, skip pandas tests')
22+
1823

1924
class LocalTargetTest(unittest.TestCase):
2025
def setUp(self):

0 commit comments

Comments
 (0)