From 56b7855f8d287f8dd3134b769a9476d9c0f5c9c3 Mon Sep 17 00:00:00 2001 From: Anushan Fernando Date: Fri, 3 Jan 2025 04:39:08 -0800 Subject: [PATCH] Potential fix for flaky ion_cyclotron_source test. This removes the default whitespaces from the seperators which will hopefully fix the flaky read. PiperOrigin-RevId: 711712482 --- torax/sources/tests/ion_cyclotron_source.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torax/sources/tests/ion_cyclotron_source.py b/torax/sources/tests/ion_cyclotron_source.py index 241eded1..87beb55c 100644 --- a/torax/sources/tests/ion_cyclotron_source.py +++ b/torax/sources/tests/ion_cyclotron_source.py @@ -91,7 +91,7 @@ def setUpClass(cls): ) # pylint: enable=protected-access with open(_DUMMY_MODEL_PATH, "w") as f: - json.dump(model_config, f) + json.dump(model_config, f, indent=4, separators=(",", ":")) cls.dummy_input = model_input cls.dummy_output = model_output super().setUpClass(