rbbt-util 5.28.10 → 5.29.1
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/lib/rbbt/hpc.rb +1 -549
- data/lib/rbbt/hpc/orchestrate.rb +24 -0
- data/lib/rbbt/hpc/slurm.rb +570 -0
- data/lib/rbbt/persist.rb +8 -3
- data/lib/rbbt/resource.rb +12 -6
- data/lib/rbbt/resource/path.rb +1 -1
- data/lib/rbbt/tsv/attach.rb +7 -4
- data/lib/rbbt/tsv/parallel.rb +0 -3
- data/lib/rbbt/util/R.rb +2 -2
- data/lib/rbbt/util/cmd.rb +9 -0
- data/lib/rbbt/util/misc/indiferent_hash.rb +8 -0
- data/lib/rbbt/util/misc/inspect.rb +23 -9
- data/lib/rbbt/workflow.rb +2 -1
- data/lib/rbbt/workflow/accessor.rb +8 -2
- data/lib/rbbt/workflow/definition.rb +1 -0
- data/lib/rbbt/workflow/examples.rb +2 -2
- data/lib/rbbt/workflow/step.rb +12 -6
- data/lib/rbbt/workflow/step/accessor.rb +47 -27
- data/lib/rbbt/workflow/step/dependencies.rb +9 -4
- data/lib/rbbt/workflow/step/run.rb +22 -20
- data/lib/rbbt/workflow/util/orchestrator.rb +14 -9
- data/lib/rbbt/workflow/util/provenance.rb +12 -5
- data/share/rbbt_commands/slurm/list +141 -0
- data/share/rbbt_commands/slurm/orchestrate +47 -0
- data/share/rbbt_commands/{workflow/slurm → slurm/task} +10 -3
- data/share/rbbt_commands/system/status +22 -22
- data/share/rbbt_commands/workflow/info +12 -9
- data/share/rbbt_commands/workflow/prov +2 -1
- data/test/rbbt/test_workflow.rb +36 -4
- data/test/rbbt/tsv/test_attach.rb +86 -6
- metadata +7 -3
@@ -17,6 +17,7 @@ $ rbbt workflow info <job-result>
|
|
17
17
|
-a--all Print all info entries
|
18
18
|
-r--recursive Print recursive input values
|
19
19
|
-o--original Print original object
|
20
|
+
-w--width* Screen width
|
20
21
|
EOF
|
21
22
|
|
22
23
|
SOPT.usage if options[:help]
|
@@ -24,6 +25,7 @@ SOPT.usage if options[:help]
|
|
24
25
|
file = ARGV.shift
|
25
26
|
all = options.delete :all
|
26
27
|
recursive = options.delete :recursive
|
28
|
+
width = (options.delete(:width) || 80).to_i
|
27
29
|
|
28
30
|
def get_step(file)
|
29
31
|
file = file.sub(/\.(info|files)/,'')
|
@@ -86,6 +88,7 @@ pid = info[:pid]
|
|
86
88
|
exception = info[:exception]
|
87
89
|
rest = info.keys - [:inputs, :dependencies, :status, :time_elapsed, :messages, :backtrace, :exception, :pid, :archived_info]
|
88
90
|
|
91
|
+
|
89
92
|
puts Log.color(:magenta, "File") << ": " << step.path
|
90
93
|
puts Log.color(:magenta, "Status") << ": " << status_msg(status) << ((step.aborted? || step.error?) && step.recoverable_error? ? " (recoverable)" : "" ) << (step.dirty? ? " (dirty)" : "")
|
91
94
|
puts Log.color(:magenta, "Pid") << ": " << pid_msg(pid, status.to_s == "done")
|
@@ -96,14 +99,14 @@ if inputs and inputs.any?
|
|
96
99
|
inputs.each do |input,value|
|
97
100
|
case value
|
98
101
|
when nil
|
99
|
-
puts Misc.format_definition_list_item(" " + input.to_s, 'nil',
|
102
|
+
puts Misc.format_definition_list_item(" " + input.to_s, 'nil', width, 20, :blue)
|
100
103
|
when Array
|
101
|
-
puts Misc.format_definition_list_item(" " + input.to_s, (value.length > 6 ? value[0..5]*"\n" << "\n" << "..." : value * "\n" ),
|
104
|
+
puts Misc.format_definition_list_item(" " + input.to_s, (value.length > 6 ? value[0..5]*"\n" << "\n" << "..." : value * "\n" ), width, 20, :blue)
|
102
105
|
when TrueClass, FalseClass
|
103
|
-
puts Misc.format_definition_list_item(" " + input.to_s, value.to_s,
|
106
|
+
puts Misc.format_definition_list_item(" " + input.to_s, value.to_s, width, 20, :blue)
|
104
107
|
else
|
105
108
|
text = value.to_s.split("\n")[0..5].compact * "\n\n"
|
106
|
-
puts Misc.format_definition_list_item(" " + input.to_s, text,
|
109
|
+
puts Misc.format_definition_list_item(" " + input.to_s, text, width, 20, :blue)
|
107
110
|
end
|
108
111
|
end
|
109
112
|
end
|
@@ -162,16 +165,16 @@ if recursive
|
|
162
165
|
inputs.each do |input,value|
|
163
166
|
case value
|
164
167
|
when nil
|
165
|
-
puts Misc.format_definition_list_item(" " << input.to_s, 'nil',
|
168
|
+
puts Misc.format_definition_list_item(" " << input.to_s, 'nil', width, 20, :blue)
|
166
169
|
when Array
|
167
|
-
puts Misc.format_definition_list_item(" " << input.to_s, (value.length > 6 ? (value[0..5])*"\n\n" << "\n\n" << "..." : value * "\n\n" ),
|
170
|
+
puts Misc.format_definition_list_item(" " << input.to_s, (value.length > 6 ? (value[0..5])*"\n\n" << "\n\n" << "..." : value * "\n\n" ), width, 20, :blue).gsub("\n\n","\n")
|
168
171
|
when TrueClass, FalseClass
|
169
|
-
puts Misc.format_definition_list_item(" " << input.to_s, value.to_s,
|
172
|
+
puts Misc.format_definition_list_item(" " << input.to_s, value.to_s, width, 20, :blue)
|
170
173
|
else
|
171
|
-
lines = value.to_s.split("\n").collect{|l| l.length >=
|
174
|
+
lines = value.to_s.split("\n").collect{|l| l.length >= width - 5 ? l[0..width - 5] + " ..." : l }
|
172
175
|
text = lines[0..5].compact * "\n\n"
|
173
176
|
text << "\n\n...\n\n" if lines.length > 6
|
174
|
-
puts Misc.format_definition_list_item(" " << input.to_s, text,
|
177
|
+
puts Misc.format_definition_list_item(" " << input.to_s, text, width, 20, :blue).gsub("\n\n","\n")
|
175
178
|
end
|
176
179
|
end
|
177
180
|
end
|
@@ -24,6 +24,7 @@ $ rbbt workflow prov <job-result>
|
|
24
24
|
-i--inputs* List of inputs to print
|
25
25
|
-if--info_fields* List of info fields to print
|
26
26
|
-t--touch Update modification times to be consistent
|
27
|
+
-e--expand_repeats Show all the dependency tree even if reapeated dependencies have already been seen before
|
27
28
|
EOF
|
28
29
|
|
29
30
|
SOPT.usage if options[:help]
|
@@ -130,6 +131,6 @@ if options[:plot]
|
|
130
131
|
end
|
131
132
|
|
132
133
|
else
|
133
|
-
puts Step.prov_report(step).strip
|
134
|
+
puts Step.prov_report(step, 0, nil, [], options[:expand_repeats]).strip
|
134
135
|
end
|
135
136
|
|
data/test/rbbt/test_workflow.rb
CHANGED
@@ -156,6 +156,15 @@ for this dependency
|
|
156
156
|
Open.read(file).reverse
|
157
157
|
end
|
158
158
|
|
159
|
+
task :create_file => :text do |file|
|
160
|
+
Open.write(file('a'), "A")
|
161
|
+
Open.write(file('b'), "B")
|
162
|
+
"DONE"
|
163
|
+
end
|
164
|
+
|
165
|
+
dep_task :reverse_step_file, TestWF, :reverse_file do |jobname, options, dependencies|
|
166
|
+
dep = dependencies.flatten.first
|
167
|
+
end
|
159
168
|
|
160
169
|
end
|
161
170
|
|
@@ -377,9 +386,9 @@ class TestWorkflow < Test::Unit::TestCase
|
|
377
386
|
job.run
|
378
387
|
Misc.with_env "RBBT_UPDATE", 'true' do
|
379
388
|
assert job.checks.select{|d| d.task_name.to_s == "t1" }.any?
|
380
|
-
job = TestWF.job(:t3)
|
381
|
-
job.step(:t1).clean
|
382
|
-
assert job.checks.select{|d| d.task_name.to_s == "t1" }.empty?
|
389
|
+
#job = TestWF.job(:t3)
|
390
|
+
#job.step(:t1).clean
|
391
|
+
#assert job.checks.select{|d| d.task_name.to_s == "t1" }.empty?
|
383
392
|
job = TestWF.job(:t3).recursive_clean
|
384
393
|
job.run
|
385
394
|
assert job.checks.select{|d| d.task_name.to_s == "t1" }.any?
|
@@ -418,7 +427,7 @@ class TestWorkflow < Test::Unit::TestCase
|
|
418
427
|
TmpFile.with_file do |dir|
|
419
428
|
Path.setup(dir)
|
420
429
|
Step.save_job_inputs(job, dir)
|
421
|
-
assert_equal Dir.glob(dir + "/*"), [dir.file.find + '.
|
430
|
+
assert_equal Dir.glob(dir + "/*"), [dir.file.find + '.yaml']
|
422
431
|
inputs = Workflow.load_inputs(dir, [:file], :file => :file)
|
423
432
|
assert_equal inputs, {:file => 'code'}
|
424
433
|
end
|
@@ -440,4 +449,27 @@ class TestWorkflow < Test::Unit::TestCase
|
|
440
449
|
end
|
441
450
|
end
|
442
451
|
end
|
452
|
+
|
453
|
+
def test_input_step_file_check
|
454
|
+
job = TestWF.job(:t3).recursive_clean
|
455
|
+
job.run
|
456
|
+
Misc.with_env "RBBT_UPDATE", 'true' do
|
457
|
+
assert job.checks.select{|d| d.task_name.to_s == "t1" }.any?
|
458
|
+
#job = TestWF.job(:t3)
|
459
|
+
#job.step(:t1).clean
|
460
|
+
#assert job.checks.select{|d| d.task_name.to_s == "t1" }.empty?
|
461
|
+
job = TestWF.job(:t3).recursive_clean
|
462
|
+
job.run
|
463
|
+
assert job.checks.select{|d| d.task_name.to_s == "t1" }.any?
|
464
|
+
job = TestWF.job(:t3)
|
465
|
+
sleep 1
|
466
|
+
Open.touch job.step(:t1).path
|
467
|
+
Misc.with_env "RBBT_UPDATE", "false" do
|
468
|
+
assert job.updated?
|
469
|
+
end
|
470
|
+
Misc.with_env "RBBT_UPDATE", "true" do
|
471
|
+
assert ! job.updated?
|
472
|
+
end
|
473
|
+
end
|
474
|
+
end
|
443
475
|
end
|
@@ -566,7 +566,7 @@ row2,CC
|
|
566
566
|
tsv3.keys.each{|k| tsv3[k] = nil if tsv3[k] == ""}
|
567
567
|
end
|
568
568
|
|
569
|
-
assert_equal tsv1.attach(tsv2, :complete => true).attach(tsv3, :complete => true)["row1"]
|
569
|
+
assert_equal [nil, "B", nil], tsv1.attach(tsv2, :complete => true).attach(tsv3, :complete => true)["row1"]
|
570
570
|
end
|
571
571
|
|
572
572
|
def test_attach_index_both_non_key
|
@@ -597,10 +597,10 @@ A Id3
|
|
597
597
|
|
598
598
|
tsv1 = tsv2 = nil
|
599
599
|
|
600
|
-
tsv1 = Rbbt.tmp.test.test1.data.tsv :double, :sep => /\s+/
|
601
|
-
tsv2 = Rbbt.tmp.test.test2.data.tsv :double, :sep => /\s+/
|
600
|
+
tsv1 = Rbbt.tmp.test.test1.data.produce(true).tsv :double, :sep => /\s+/
|
601
|
+
tsv2 = Rbbt.tmp.test.test2.data.produce(true).tsv :double, :sep => /\s+/
|
602
602
|
|
603
|
-
tsv2.identifiers = Rbbt.tmp.test.test2.identifiers.produce.find #.to_s
|
603
|
+
tsv2.identifiers = Rbbt.tmp.test.test2.identifiers.produce(true).produce.find #.to_s
|
604
604
|
|
605
605
|
tsv1.attach tsv2, :fields => ["ValueE"] #, :persist_input => true
|
606
606
|
Log.tsv tsv1
|
@@ -627,12 +627,92 @@ E B
|
|
627
627
|
|
628
628
|
tsv1 = tsv2 = nil
|
629
629
|
|
630
|
-
tsv1 = Rbbt.tmp.test.test1.data.tsv :double, :sep => /\s+/
|
631
|
-
tsv2 = Rbbt.tmp.test.test2.data.tsv :double, :sep => /\s+/
|
630
|
+
tsv1 = Rbbt.tmp.test.test1.data.produce(true).tsv :double, :sep => /\s+/
|
631
|
+
tsv2 = Rbbt.tmp.test.test2.data.produce(true).tsv :double, :sep => /\s+/
|
632
632
|
|
633
633
|
tsv1.attach tsv2, :fields => ["ValueE"] #, :persist_input => true
|
634
634
|
Log.tsv tsv1
|
635
635
|
|
636
636
|
end
|
637
|
+
|
638
|
+
def test_attach_complete
|
639
|
+
content1 =<<-EOF
|
640
|
+
#: :sep=/\\s+/
|
641
|
+
#Id ValueA
|
642
|
+
row1 a|aa|aaa
|
643
|
+
row2 A
|
644
|
+
EOF
|
645
|
+
|
646
|
+
content2 =<<-EOF
|
647
|
+
#: :sep=/\\s+/
|
648
|
+
#Id ValueB
|
649
|
+
row1 b
|
650
|
+
row3 C
|
651
|
+
EOF
|
652
|
+
Rbbt.claim Rbbt.tmp.test.test1.data, :string, content1
|
653
|
+
Rbbt.claim Rbbt.tmp.test.test2.data, :string, content2
|
654
|
+
|
655
|
+
tsv1 = tsv2 = nil
|
656
|
+
|
657
|
+
tsv1 = Rbbt.tmp.test.test1.data.produce(true).tsv :double, :sep => /\s+/
|
658
|
+
tsv2 = Rbbt.tmp.test.test2.data.produce(true).tsv :double, :sep => /\s+/
|
659
|
+
|
660
|
+
tsv1.attach tsv2, :complete => true
|
661
|
+
assert_equal [[], ["C"]], tsv1["row3"]
|
662
|
+
|
663
|
+
tsv1 = Rbbt.tmp.test.test1.data.produce(true).tsv :double, :sep => /\s+/
|
664
|
+
tsv2 = Rbbt.tmp.test.test2.data.produce(true).tsv :double, :sep => /\s+/
|
665
|
+
|
666
|
+
ppp tsv1.attach tsv2, :complete => ["AA"]
|
667
|
+
tsv1.attach tsv2, :complete => ["AA"]
|
668
|
+
assert_equal [["AA"], ["C"]], tsv1["row3"]
|
669
|
+
end
|
670
|
+
|
671
|
+
def test_attach_complete_identifiers
|
672
|
+
content1 =<<-EOF
|
673
|
+
#: :sep=/\\s+/
|
674
|
+
#Id ValueA
|
675
|
+
row1 a|aa|aaa
|
676
|
+
row2 A
|
677
|
+
EOF
|
678
|
+
|
679
|
+
content2 =<<-EOF
|
680
|
+
#: :sep=/\\s+/
|
681
|
+
#Id2 ValueB
|
682
|
+
ROW_1 b
|
683
|
+
ROW_2 C
|
684
|
+
EOF
|
685
|
+
|
686
|
+
identifiers =<<-EOF
|
687
|
+
#: :sep=/\\s+/
|
688
|
+
#Id Id2
|
689
|
+
row1 ROW_1
|
690
|
+
row2 ROW_2
|
691
|
+
row3 ROW_3
|
692
|
+
EOF
|
693
|
+
Rbbt.claim Rbbt.tmp.test.test1.data, :string, content1
|
694
|
+
Rbbt.claim Rbbt.tmp.test.test2.data, :string, content2
|
695
|
+
Rbbt.claim Rbbt.tmp.test.identifiers.data, :string, identifiers
|
696
|
+
|
697
|
+
tsv1 = tsv2 = nil
|
698
|
+
|
699
|
+
tsv1 = Rbbt.tmp.test.test1.data.produce(true).tsv :double, :sep => /\s+/
|
700
|
+
tsv2 = Rbbt.tmp.test.test2.data.produce(true).tsv :double, :sep => /\s+/
|
701
|
+
ids = Rbbt.tmp.test.identifiers.data.produce(true).tsv :double, :sep => /\s+/
|
702
|
+
|
703
|
+
tsv1.identifiers = ids
|
704
|
+
|
705
|
+
tsv1.attach tsv2
|
706
|
+
assert_equal [["A"], ["C"]], tsv1["row2"]
|
707
|
+
|
708
|
+
tsv1 = Rbbt.tmp.test.test1.data.produce(true).tsv :double, :sep => /\s+/
|
709
|
+
tsv2 = Rbbt.tmp.test.test2.data.produce(true).tsv :double, :sep => /\s+/
|
710
|
+
ids = Rbbt.tmp.test.identifiers.data.produce(true).tsv :double, :sep => /\s+/
|
711
|
+
|
712
|
+
tsv1.identifiers = ids
|
713
|
+
|
714
|
+
tsv1.attach tsv2, :complete => true
|
715
|
+
assert_equal [["A"], ["C"]], tsv1["row2"]
|
716
|
+
end
|
637
717
|
end
|
638
718
|
|
metadata
CHANGED
@@ -1,14 +1,14 @@
|
|
1
1
|
--- !ruby/object:Gem::Specification
|
2
2
|
name: rbbt-util
|
3
3
|
version: !ruby/object:Gem::Version
|
4
|
-
version: 5.
|
4
|
+
version: 5.29.1
|
5
5
|
platform: ruby
|
6
6
|
authors:
|
7
7
|
- Miguel Vazquez
|
8
8
|
autorequire:
|
9
9
|
bindir: bin
|
10
10
|
cert_chain: []
|
11
|
-
date:
|
11
|
+
date: 2021-01-14 00:00:00.000000000 Z
|
12
12
|
dependencies:
|
13
13
|
- !ruby/object:Gem::Dependency
|
14
14
|
name: rake
|
@@ -211,6 +211,8 @@ files:
|
|
211
211
|
- lib/rbbt/entity/identifiers.rb
|
212
212
|
- lib/rbbt/fix_width_table.rb
|
213
213
|
- lib/rbbt/hpc.rb
|
214
|
+
- lib/rbbt/hpc/orchestrate.rb
|
215
|
+
- lib/rbbt/hpc/slurm.rb
|
214
216
|
- lib/rbbt/knowledge_base.rb
|
215
217
|
- lib/rbbt/knowledge_base/enrichment.rb
|
216
218
|
- lib/rbbt/knowledge_base/entity.rb
|
@@ -376,6 +378,9 @@ files:
|
|
376
378
|
- share/rbbt_commands/resource/produce
|
377
379
|
- share/rbbt_commands/resource/read
|
378
380
|
- share/rbbt_commands/rsync
|
381
|
+
- share/rbbt_commands/slurm/list
|
382
|
+
- share/rbbt_commands/slurm/orchestrate
|
383
|
+
- share/rbbt_commands/slurm/task
|
379
384
|
- share/rbbt_commands/stat/abs
|
380
385
|
- share/rbbt_commands/stat/boxplot
|
381
386
|
- share/rbbt_commands/stat/compare_lists
|
@@ -430,7 +435,6 @@ files:
|
|
430
435
|
- share/rbbt_commands/workflow/remote/list
|
431
436
|
- share/rbbt_commands/workflow/remote/remove
|
432
437
|
- share/rbbt_commands/workflow/server
|
433
|
-
- share/rbbt_commands/workflow/slurm
|
434
438
|
- share/rbbt_commands/workflow/task
|
435
439
|
- share/rbbt_commands/workflow/trace
|
436
440
|
- share/unicorn.rb
|