@@ -20,10 +20,7 @@ runs the scheduler with the specified options. Returns a Chunk which references
20
20
the result.
21
21
"""
22
22
function compute (ctx:: Context , d:: Thunk ; options= nothing )
23
- scheduler = get! (PLUGINS, :scheduler ) do
24
- get_type (PLUGIN_CONFIGS[:scheduler ])
25
- end
26
- res = scheduler. compute_dag (ctx, d; options= options)
23
+ result = Sch. compute_dag (ctx, d; options= options)
27
24
if ctx. log_file != = nothing
28
25
if ctx. log_sink isa TimespanLogging. LocalEventLog
29
26
logs = TimespanLogging. get_logs! (ctx. log_sink)
@@ -34,12 +31,12 @@ function compute(ctx::Context, d::Thunk; options=nothing)
34
31
@warn " Context log_sink not set to LocalEventLog, skipping"
35
32
end
36
33
end
37
- res
34
+ result
38
35
end
39
36
40
37
function debug_compute (ctx:: Context , args... ; profile= false , options= nothing )
41
- @time res = compute (ctx, args... ; options= options)
42
- get_logs! (ctx. log_sink), res
38
+ @time result = compute (ctx, args... ; options= options)
39
+ get_logs! (ctx. log_sink), result
43
40
end
44
41
45
42
function debug_compute (arg; profile= false , options= nothing )
@@ -53,11 +50,7 @@ Base.@deprecate gather(x) collect(x)
53
50
54
51
cleanup () = cleanup (Context (global_context ()))
55
52
function cleanup (ctx:: Context )
56
- if :scheduler in keys (PLUGINS)
57
- scheduler = PLUGINS[:scheduler ]
58
- (scheduler). cleanup (ctx)
59
- delete! (PLUGINS, :scheduler )
60
- end
53
+ Sch. cleanup (ctx)
61
54
nothing
62
55
end
63
56
0 commit comments