Rename Torchserve image integration test.

IQTLabs · Feb 2, 2024 · 1509085 · 1509085
1 parent 4846bfd
commit 1509085
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 1 deletion.
diff --git a/tests/test_torchserve.sh → tests/test_image_torchserve.sh b/tests/test_torchserve.sh → tests/test_image_torchserve.sh
@@ -4,7 +4,7 @@ set -e
 TMPDIR=/tmp
 sudo apt-get update && sudo apt-get install -y jq wget
 sudo pip3 install torch-model-archiver
-cp torchserve/* $TMPDIR/
+cp torchserve/image/* $TMPDIR/
 cd $TMPDIR
 # TODO: use gamutRF weights here.
 wget https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8n.pt

diff --git a/torchserve/custom_handler.py → torchserve/image/custom_handler.py b/torchserve/custom_handler.py → torchserve/image/custom_handler.py
diff --git a/torchserve/model_config.json → torchserve/image/model_config.json b/torchserve/model_config.json → torchserve/image/model_config.json