Skip to content

Commit 3eb97a8

Browse files
author
Frankie Robertson
committed
Refactor compatible_tracker a bit
1 parent ea8b4e0 commit 3eb97a8

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

src/aggregators/Aggregators.jl

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -91,12 +91,16 @@ function AbilityTracker(bits...; integrator = nothing, ability_estimator = nothi
9191
end
9292
end
9393

94-
function compatible_tracker(bits...; integrator, ability_estimator, prefer_tracked)
95-
ability_tracker = AbilityTracker(bits...; ability_estimator = ability_estimator)
96-
if ability_tracker isa GriddedAbilityTracker &&
94+
function find_ability_tracker(ability_tracker, typ, integrator)
95+
if ability_tracker isa typ &&
9796
ability_tracker.integrator === integrator
9897
return ability_tracker
9998
end
99+
end
100+
101+
function compatible_tracker(bits...; integrator, ability_estimator, prefer_tracked)
102+
ability_tracker = AbilityTracker(bits...; ability_estimator = ability_estimator)
103+
@returnsome find_ability_tracker(ability_tracker, GriddedAbilityTracker, integrator)
100104
if prefer_tracked
101105
return AbilityTracker(bits...;
102106
integrator = integrator,

0 commit comments

Comments
 (0)