diff --git a/tika-parsers/tika-parsers-ml/tika-dl/src/test/java/org/apache/tika/dl/imagerec/DL4JInceptionV3NetTest.java b/tika-parsers/tika-parsers-ml/tika-dl/src/test/java/org/apache/tika/dl/imagerec/DL4JInceptionV3NetTest.java index 0fb8fc0ba6..2d19d1d22b 100644 --- a/tika-parsers/tika-parsers-ml/tika-dl/src/test/java/org/apache/tika/dl/imagerec/DL4JInceptionV3NetTest.java +++ b/tika-parsers/tika-parsers-ml/tika-dl/src/test/java/org/apache/tika/dl/imagerec/DL4JInceptionV3NetTest.java @@ -39,7 +39,7 @@ public void recognise() throws Exception { config = new TikaConfig(is); } catch (Exception e) { if (e.getMessage() != null && (e.getMessage().contains("Connection refused") || - e.getMessage().contains("connect timed out"))) { + e.getMessage().contains("connect timed out") || e.getMessage().contains("404"))) { assumeTrue(false, "skipping test because of connection issue"); } throw e; diff --git a/tika-parsers/tika-parsers-ml/tika-dl/src/test/java/org/apache/tika/dl/imagerec/DL4JVGG16NetTest.java b/tika-parsers/tika-parsers-ml/tika-dl/src/test/java/org/apache/tika/dl/imagerec/DL4JVGG16NetTest.java index fcf157eb77..83ce2e9909 100644 --- a/tika-parsers/tika-parsers-ml/tika-dl/src/test/java/org/apache/tika/dl/imagerec/DL4JVGG16NetTest.java +++ b/tika-parsers/tika-parsers-ml/tika-dl/src/test/java/org/apache/tika/dl/imagerec/DL4JVGG16NetTest.java @@ -39,7 +39,7 @@ public void recognise() throws Exception { config = new TikaConfig(is); } catch (Exception e) { if (e.getMessage() != null && (e.getMessage().contains("Connection refused") || - e.getMessage().contains("connect timed out"))) { + e.getMessage().contains("connect timed out") || e.getMessage().contains("404"))) { assumeTrue(false, "skipping test because of connection issue"); } throw e;