Skip to content

Commit 73b60f8

Browse files
committed
Update TorchServe image name to fix CI test
1 parent 04c1946 commit 73b60f8

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/test/java/com/github/tadayosi/torchserve/client/ManagementTest.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ void testListModels() throws Exception {
136136
@Test
137137
void testSetDefault() throws Exception {
138138
var response = client.management().setDefault(DEFAULT_MODEL, DEFAULT_MODEL_VERSION);
139-
assertTrue(response.getStatus().contains("Default vesion succsesfully updated"));
139+
assertTrue(response.getStatus().contains("Default version successfully updated"));
140140
}
141141

142142
@Test

src/test/java/com/github/tadayosi/torchserve/client/TorchServeTestSupport.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010
class TorchServeTestSupport {
1111

12-
private static final String IMAGE_NAME = "aarch64".equals(System.getProperty("os.arch")) ? "tadayosi/torchserve" : "pytorch/torchserve";
12+
private static final String IMAGE_NAME = "pytorch/torchserve";
1313

1414
@Container
1515
static GenericContainer<?> torchServe = new GenericContainer<>(DockerImageName.parse(IMAGE_NAME))

0 commit comments

Comments
 (0)