Skip to content

Commit a53d414

Browse files
committed
fix tests
1 parent c55c3ab commit a53d414

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

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
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)