File tree Expand file tree Collapse file tree 2 files changed +19
-1
lines changed Expand file tree Collapse file tree 2 files changed +19
-1
lines changed Original file line number Diff line number Diff line change 23
23
24
24
sanity_check :
25
25
parameters :
26
- inputs : {data_path: data/, parameters_file: {type: file, default: parameters.yaml}}
26
+ inputs : {data_path: data/, parameters_file: {type: file, default: parameters.yaml}}
27
+
28
+ statistics :
29
+ parameters :
30
+ inputs : {data_path: data/, parameters_file: {type: file, default: parameters.yaml}}
31
+ outputs : {output_path: {type: file, default: statistics.yaml}}
Original file line number Diff line number Diff line change 1
1
"""MLCube handler file"""
2
2
import os
3
3
import typer
4
+ import yaml
4
5
import subprocess
5
6
6
7
@@ -43,6 +44,18 @@ def sanity_check(
43
44
):
44
45
pass
45
46
47
+ @app .command ("statistics" )
48
+ def statistics (
49
+ data_path : str = typer .Option (..., "--data_path" ),
50
+ parameters_file : str = typer .Option (..., "--parameters_file" ),
51
+ output_path : str = typer .Option (..., "--output_path" )
52
+ ):
53
+ stats = {
54
+ "stat" : 1
55
+ }
56
+ with open (output_path , "w" ) as f :
57
+ yaml .dump (stats , f )
58
+
46
59
47
60
if __name__ == "__main__" :
48
61
app ()
You can’t perform that action at this time.
0 commit comments