change pt->torch_script

Signed-off-by: HenryL27 <[email protected]>
HenryL27 · Dec 6, 2023 · be113ed · be113ed
1 parent b3e90b5
commit be113ed
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.../opensearch/ml/engine/algorithms/text_similarity/TextSimilarityCrossEncoderModelTest.java b/.../opensearch/ml/engine/algorithms/text_similarity/TextSimilarityCrossEncoderModelTest.java
@@ -98,7 +98,7 @@ public void setUp() throws URISyntaxException {
             .build();
         modelHelper = new ModelHelper(mlEngine);
         params = new HashMap<>();
-        modelZipFile = new File(getClass().getResource("TinyBERT-CE-pt.zip").toURI());
+        modelZipFile = new File(getClass().getResource("TinyBERT-CE-torch_script.zip").toURI());
         params.put(MODEL_ZIP_FILE, modelZipFile);
         params.put(MODEL_HELPER, modelHelper);
         params.put(ML_ENGINE, mlEngine);
@@ -220,7 +220,7 @@ public void initModel_predict_ONNX_CrossEncoder() throws URISyntaxException {
     @Test
     public void initModel_NullModelHelper() throws URISyntaxException {
         Map<String, Object> params = new HashMap<>();
-        params.put(MODEL_ZIP_FILE, new File(getClass().getResource("TinyBERT-CE-pt.zip").toURI()));
+        params.put(MODEL_ZIP_FILE, new File(getClass().getResource("TinyBERT-CE-torch_script.zip").toURI()));
         IllegalArgumentException e = assertThrows(
             IllegalArgumentException.class,
             () -> textSimilarityCrossEncoderModel.initModel(model, params, encryptor)
@@ -231,7 +231,7 @@ public void initModel_NullModelHelper() throws URISyntaxException {
     @Test
     public void initModel_NullMLEngine() throws URISyntaxException {
         Map<String, Object> params = new HashMap<>();
-        params.put(MODEL_ZIP_FILE, new File(getClass().getResource("TinyBERT-CE-pt.zip").toURI()));
+        params.put(MODEL_ZIP_FILE, new File(getClass().getResource("TinyBERT-CE-torch_script.zip").toURI()));
         params.put(MODEL_HELPER, modelHelper);
         IllegalArgumentException e = assertThrows(
             IllegalArgumentException.class,

diff --git a/...rithms/text_similarity/TinyBERT-CE-pt.zip → ...t_similarity/TinyBERT-CE-torch_script.zip b/...rithms/text_similarity/TinyBERT-CE-pt.zip → ...t_similarity/TinyBERT-CE-torch_script.zip