@@ -31,9 +31,13 @@ def __init__(self, data_requirement=None, server=False):
31
31
self ._data_requirement_path = None
32
32
self ._data_requirement_file = None
33
33
self ._use_server = server
34
+ self ._server_dst_folder = "./data"
34
35
35
36
self .load_data_requirement (data_requirement )
36
37
38
+ def set_server_dst_folder (self ,directory ):
39
+ self ._server_dst_folder = directory
40
+
37
41
def _set_data_requirement_path (self , data_requirement_path ):
38
42
"""Define path to the requirement file"""
39
43
if data_requirement_path is None :
@@ -99,19 +103,19 @@ def install(self):
99
103
for key , value in self ._data_requirement_file .items ():
100
104
if isinstance (value , dict ):
101
105
ret += [Repo2DataChild (value , self ._use_server ,
102
- self ._data_requirement_path ,key ).install ()]
106
+ self ._data_requirement_path ,key , self . _server_dst_folder ).install ()]
103
107
# if not, it is a single assignment
104
108
else :
105
109
ret += [Repo2DataChild (self ._data_requirement_file ,
106
- self ._use_server , self ._data_requirement_path ).install ()]
110
+ self ._use_server , self ._data_requirement_path , None , self . _server_dst_folder ).install ()]
107
111
108
112
return ret
109
113
110
114
111
115
class Repo2DataChild ():
112
116
"""Repo2data child class which install the dataset"""
113
117
114
- def __init__ (self , data_requirement_file = None , use_server = False , data_requirement_path = None , download_key = None ):
118
+ def __init__ (self , data_requirement_file = None , use_server = False , data_requirement_path = None , download_key = None , server_dst_folder = None ):
115
119
"""Initialize the Repo2Data child class.
116
120
Parameters
117
121
----------
@@ -124,7 +128,7 @@ def __init__(self, data_requirement_file=None, use_server=False, data_requiremen
124
128
self ._dst_path = None
125
129
self ._use_server = use_server
126
130
self ._data_requirement_path = data_requirement_path
127
- self ._server_dst_folder = "./data"
131
+ self ._server_dst_folder = server_dst_folder
128
132
self ._download_key = download_key
129
133
if self ._download_key :
130
134
self ._cache_record = f"{ self ._download_key } _repo2data_cache_record.json"
@@ -133,9 +137,6 @@ def __init__(self, data_requirement_file=None, use_server=False, data_requiremen
133
137
134
138
self .load_data_requirement (data_requirement_file )
135
139
136
- def set_server_dst_folder (self ,directory ):
137
- self ._server_dst_folder = directory
138
-
139
140
def load_data_requirement (self , data_requirement_file ):
140
141
"""Load the json data requirement file and set destination folder"""
141
142
# here we should load just a json data
0 commit comments