static DLModels

Signed-off-by: HenryL27 <[email protected]>
HenryL27 · Dec 5, 2023 · 97e10b1 · 97e10b1
1 parent 6d7b429
commit 97e10b1
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/common/src/main/java/org/opensearch/ml/common/FunctionName.java b/common/src/main/java/org/opensearch/ml/common/FunctionName.java
@@ -34,17 +34,18 @@ public static FunctionName from(String value) {
         }
     }
 
+    private static final HashSet<FunctionName> DL_MODELS = new HashSet<>(Set.of(
+        TEXT_EMBEDDING,
+        TEXT_SIMILARITY,
+        SPARSE_ENCODING,
+        SPARSE_TOKENIZE
+    ));
+
     /**
      * Check if model is deep learning model.
      * @return true for deep learning model.
      */
     public static boolean isDLModel(FunctionName functionName) {
-        Set<FunctionName> dlmodels = new HashSet<>(Set.of(
-            TEXT_EMBEDDING,
-            TEXT_SIMILARITY,
-            SPARSE_ENCODING,
-            SPARSE_TOKENIZE
-        ));
-        return dlmodels.contains(functionName);
+        return DL_MODELS.contains(functionName);
     }
 }