make enableSuite => get or create

apache · Feb 1, 2025 · 7a78eb0 · 7a78eb0
1 parent 1281743
commit 7a78eb0
Showing 1 changed file with 5 additions and 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
@@ -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 = {