From 168783d89b2c281256d179fd5e67e8831a532a05 Mon Sep 17 00:00:00 2001 From: Giovanni Iadarola Date: Tue, 7 Nov 2023 09:12:34 +0100 Subject: [PATCH] Fix in GPU test --- tests/test_tracker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_tracker.py b/tests/test_tracker.py index 33ac9984c..94e1fd388 100644 --- a/tests/test_tracker.py +++ b/tests/test_tracker.py @@ -527,7 +527,7 @@ def test_backtrack_with_flag(test_context): ) @pytest.mark.parametrize('collective', [True, False], ids=['collective', 'non-collective']) def test_tracking_with_progress(test_context, with_progress, turns, collective): - elements = [xt.Drift(length=2.) for _ in range(5)] + elements = [xt.Drift(length=2, _context=test_context) for _ in range(5)] elements[3].iscollective = collective line = xt.Line(elements=elements) line.reset_s_at_end_turn = False