From 7a78eb0521ad0b637cc500f3122514d48173ee65 Mon Sep 17 00:00:00 2001 From: Chang Chen Date: Sat, 1 Feb 2025 17:19:34 +0800 Subject: [PATCH] make enableSuite => get or create --- .../org/apache/gluten/utils/BackendTestSettings.scala | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/gluten-ut/common/src/test/scala/org/apache/gluten/utils/BackendTestSettings.scala b/gluten-ut/common/src/test/scala/org/apache/gluten/utils/BackendTestSettings.scala index 51e8174da7fb..20e3a53200a7 100644 --- a/gluten-ut/common/src/test/scala/org/apache/gluten/utils/BackendTestSettings.scala +++ b/gluten-ut/common/src/test/scala/org/apache/gluten/utils/BackendTestSettings.scala @@ -34,12 +34,12 @@ abstract class BackendTestSettings { } protected def enableSuite(suiteName: String): SuiteSettings = { - if (enabledSuites.containsKey(suiteName)) { - throw new IllegalArgumentException("Duplicated suite name: " + suiteName) + if (!enabledSuites.containsKey(suiteName)) { + val suiteSettings = new SuiteSettings + enabledSuites.put(suiteName, suiteSettings) + enabledSuites.get(suiteName) } - val suiteSettings = new SuiteSettings - enabledSuites.put(suiteName, suiteSettings) - suiteSettings + enabledSuites.get(suiteName) } private[utils] def shouldRun(suiteName: String, testName: String): Boolean = {