26
26
27
27
import click
28
28
import pendulum
29
-
30
29
from cascade .base import MetaHandler
31
30
32
31
@@ -100,13 +99,9 @@ def parse_value(value: ast.expr) -> Any:
100
99
elif isinstance (value , ast .Set ):
101
100
return set (parse_value (v ) for v in value .elts )
102
101
elif isinstance (value , ast .Dict ):
103
- return {
104
- parse_value (k ): parse_value (v ) for k , v in zip (value .keys , value .values )
105
- }
102
+ return {parse_value (k ): parse_value (v ) for k , v in zip (value .keys , value .values )}
106
103
else :
107
- raise ValueError (
108
- f"Unsupported config field type: { value } in { unparse_method (value )} "
109
- )
104
+ raise ValueError (f"Unsupported config field type: { value } in { unparse_method (value )} " )
110
105
111
106
112
107
# ast.unparse exists since python 3.9
@@ -134,9 +129,7 @@ def node2dict(cfg_node: ast.ClassDef) -> Dict[str, Any]:
134
129
return cfg_dict
135
130
136
131
137
- def modify_assignments (
138
- tree : ast .Module , cfg_node : ast .ClassDef , kwargs : Dict [str , Any ]
139
- ) -> str :
132
+ def modify_assignments (tree : ast .Module , cfg_node : ast .ClassDef , kwargs : Dict [str , Any ]) -> str :
140
133
"""
141
134
Overrides cascade.base.Config class definition with user-provided values
142
135
"""
@@ -169,9 +162,7 @@ def generate_run_id() -> str:
169
162
170
163
171
164
class CascadeRun :
172
- def __init__ (
173
- self , log : bool , config : Dict [str , Any ], overrides : Dict [str , Any ]
174
- ) -> None :
165
+ def __init__ (self , log : bool , config : Dict [str , Any ], overrides : Dict [str , Any ]) -> None :
175
166
self .log = log
176
167
self .config = config
177
168
self .overrides = overrides
@@ -184,12 +175,8 @@ def __init__(
184
175
def __enter__ (self ):
185
176
os .makedirs (self .run_dir )
186
177
187
- MetaHandler .write (
188
- os .path .join (self .run_dir , "cascade_config.json" ), self .config
189
- )
190
- MetaHandler .write (
191
- os .path .join (self .run_dir , "cascade_overrides.json" ), self .overrides
192
- )
178
+ MetaHandler .write (os .path .join (self .run_dir , "cascade_config.json" ), self .config )
179
+ MetaHandler .write (os .path .join (self .run_dir , "cascade_overrides.json" ), self .overrides )
193
180
194
181
return self
195
182
@@ -270,7 +257,6 @@ def run(script: str, y: bool, log: Optional[str], args: List[str]):
270
257
else :
271
258
raise KeyError (f"Key `{ key } ` is missing in the original config" )
272
259
273
- print ([item for item in tree .body ])
274
260
text = modify_assignments (tree , cfg_node , kwargs )
275
261
else :
276
262
cfg_dict = {}
0 commit comments