Skip to content

Commit a7dce6f

Browse files
committed
Fixed a bug
Signed-off-by: romit <romit@ibm.com>
1 parent 17bafe4 commit a7dce6f

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

tuning/data/data_processors.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -112,9 +112,9 @@ def load_dataset(
112112
if (not datafile) and (not datasetconfig):
113113
raise ValueError("Either datafile or datasetconfig must be set")
114114

115-
effective_split = (
116-
splitName or datasetconfig.dataset_split_name if datasetconfig else None
117-
)
115+
effective_split = splitName
116+
if datasetconfig and datasetconfig.dataset_split_name:
117+
effective_split = datasetconfig.dataset_split_name
118118

119119
def _load_dataset(
120120
data_path=None,

0 commit comments

Comments
 (0)