Skip to content

Commit 48d4ef7

Browse files
committed
[DLMED] fix CI tests
Signed-off-by: Nic Ma <nma@nvidia.com>
1 parent 1903529 commit 48d4ef7

File tree

2 files changed

+7
-9
lines changed

2 files changed

+7
-9
lines changed

monai/data/utils.py

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -38,13 +38,8 @@
3838
)
3939
from monai.utils.enums import Method
4040

41-
if TYPE_CHECKING:
42-
import pandas as pd
43-
from pandas import DataFrame
44-
else:
45-
pd, _ = optional_import("pandas")
46-
DataFrame, _ = optional_import("pandas", name="DataFrame")
47-
41+
pd, _ = optional_import("pandas")
42+
DataFrame, _ = optional_import("pandas", name="DataFrame")
4843
nib, _ = optional_import("nibabel")
4944

5045

@@ -996,7 +991,7 @@ def sorted_dict(item, key=None, reverse=False):
996991

997992

998993
def convert_tables_to_dicts(
999-
dfs: Union[Sequence[DataFrame], DataFrame],
994+
dfs,
1000995
row_indices: Optional[Sequence[Union[int, str]]] = None,
1001996
col_names: Optional[Sequence[str]] = None,
1002997
col_groups: Optional[Dict[str, Sequence[str]]] = None,

tests/test_csv_iterable_dataset.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111

1212
import os
1313
import tempfile
14+
import sys
1415
import unittest
1516

1617
import numpy as np
@@ -160,7 +161,9 @@ def prepare_csv_file(data, filepath):
160161

161162
# test multiple processes loading
162163
dataset = CSVIterableDataset(filepath1, transform=ToNumpyd(keys="label"))
163-
dataloader = DataLoader(dataset=dataset, num_workers=2, batch_size=2)
164+
# num workers = 0 for mac
165+
num_workers = 0 if sys.platform == "darwin" else 2
166+
dataloader = DataLoader(dataset=dataset, num_workers=num_workers, batch_size=2)
164167
for item in dataloader:
165168
# test the last item which only has 1 data
166169
if len(item) == 1:

0 commit comments

Comments
 (0)