From be113eda1897bb7e04ca6423d43cc6b490388600 Mon Sep 17 00:00:00 2001 From: HenryL27 Date: Wed, 6 Dec 2023 15:51:48 -0800 Subject: [PATCH] change pt->torch_script Signed-off-by: HenryL27 --- .../TextSimilarityCrossEncoderModelTest.java | 6 +++--- ...yBERT-CE-pt.zip => TinyBERT-CE-torch_script.zip} | Bin 2 files changed, 3 insertions(+), 3 deletions(-) rename ml-algorithms/src/test/resources/org/opensearch/ml/engine/algorithms/text_similarity/{TinyBERT-CE-pt.zip => TinyBERT-CE-torch_script.zip} (100%) diff --git a/ml-algorithms/src/test/java/org/opensearch/ml/engine/algorithms/text_similarity/TextSimilarityCrossEncoderModelTest.java b/ml-algorithms/src/test/java/org/opensearch/ml/engine/algorithms/text_similarity/TextSimilarityCrossEncoderModelTest.java index bc991bd4f4..88e64e2517 100644 --- a/ml-algorithms/src/test/java/org/opensearch/ml/engine/algorithms/text_similarity/TextSimilarityCrossEncoderModelTest.java +++ b/ml-algorithms/src/test/java/org/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 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 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/ml-algorithms/src/test/resources/org/opensearch/ml/engine/algorithms/text_similarity/TinyBERT-CE-pt.zip b/ml-algorithms/src/test/resources/org/opensearch/ml/engine/algorithms/text_similarity/TinyBERT-CE-torch_script.zip similarity index 100% rename from ml-algorithms/src/test/resources/org/opensearch/ml/engine/algorithms/text_similarity/TinyBERT-CE-pt.zip rename to ml-algorithms/src/test/resources/org/opensearch/ml/engine/algorithms/text_similarity/TinyBERT-CE-torch_script.zip