Clean-up scheduler plugin accessor

JuliaParallel · Oct 29, 2020 · edcb640 · edcb640
1 parent 2aa4cc6
commit edcb640
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/compute.jl b/src/compute.jl
@@ -25,11 +25,10 @@ runs the scheduler with the specified options. Returns a Chunk which references
 the result.
 """
 function compute(ctx::Context, d::Thunk; options=nothing)
-    if !(:scheduler in keys(PLUGINS))
-        PLUGINS[:scheduler] = get_type(PLUGIN_CONFIGS[:scheduler])
+    scheduler = get!(PLUGINS, :scheduler) do
+        get_type(PLUGIN_CONFIGS[:scheduler])
     end
-    scheduler = PLUGINS[:scheduler]
-    res = (scheduler).compute_dag(ctx, d; options=options)
+    res = scheduler.compute_dag(ctx, d; options=options)
     if ctx.log_file !== nothing
         if ctx.log_sink !== LocalEventLog
             logs = get_logs!(ctx.log_sink)