Skip to content
This repository was archived by the owner on Mar 17, 2021. It is now read-only.

Commit f5446a5

Browse files
committed
set default data splitting file
1 parent 7163e46 commit f5446a5

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

niftynet/io/image_sets_partitioner.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ class ImageSetsPartitioner(object):
6161
def initialise(self,
6262
data_param,
6363
new_partition=False,
64-
data_split_file="./test.csv",
64+
data_split_file=None,
6565
ratios=None):
6666
"""
6767
Set the data partitioner parameters
@@ -77,7 +77,10 @@ def initialise(self,
7777
and get_file_list always returns all subjects.
7878
"""
7979
self.data_param = data_param
80-
self.data_split_file = data_split_file
80+
if data_split_file is None:
81+
self.data_split_file = os.path.join('.', 'dataset_split.csv')
82+
else:
83+
self.data_split_file = data_split_file
8184
self.ratios = ratios
8285

8386
self._file_list = None

0 commit comments

Comments
 (0)