File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed
Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -2489,12 +2489,15 @@ def _parse_level_list(
24892489 # NOTE: This should have no effect if levels were generated automatically.
24902490 # However want to apply these to manual-input levels as well.
24912491 def _restrict_levels (levels ):
2492+ kw = {}
2493+ if len (levels ) > 2 :
2494+ kw ['atol' ] = 1e-5 * np .min (np .diff (levels ))
24922495 if nozero :
2493- levels = levels [ levels != 0 ]
2496+ levels = [ lev for lev in levels if not np . isclose ( lev , 0 , ** kw ) ]
24942497 if positive :
2495- levels = levels [ levels >= 0 ]
2498+ levels = [ lev for lev in levels if lev > 0 or np . isclose ( lev , 0 , ** kw ) ]
24962499 if negative :
2497- levels = levels [ levels <= 0 ]
2500+ levels = [ lev for lev in levels if lev < 0 or np . isclose ( lev , 0 , ** kw ) ]
24982501 return levels
24992502
25002503 # Helper function to sanitize input levels
You can’t perform that action at this time.
0 commit comments