[api] Log warning message if failed to load specified class (deepjava…

…library#2724)
madfu · Jul 27, 2023 · ae84c25 · ae84c25
1 parent cd4f4ee
commit ae84c25
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/api/src/main/java/ai/djl/repository/zoo/BaseModelLoader.java b/api/src/main/java/ai/djl/repository/zoo/BaseModelLoader.java
@@ -30,6 +30,9 @@
 import ai.djl.util.Progress;
 import ai.djl.util.Utils;
 
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
 import java.io.IOException;
 import java.io.Reader;
 import java.lang.reflect.Type;
@@ -43,6 +46,8 @@
 /** Shared code for the {@link ModelLoader} implementations. */
 public class BaseModelLoader implements ModelLoader {
 
+    private static final Logger logger = LoggerFactory.getLogger(BaseModelLoader.class);
+
     protected MRL mrl;
     protected TranslatorFactory defaultFactory;
 
@@ -238,6 +243,9 @@ protected TranslatorFactory getTranslatorFactory(
         if (factoryClass != null) {
             ClassLoader cl = ClassLoaderUtils.getContextClassLoader();
             factory = ClassLoaderUtils.initClass(cl, TranslatorFactory.class, factoryClass);
+            if (factory == null) {
+                logger.warn("Failed to load translatorFactory: {}", factoryClass);
+            }
         }
         return factory;
     }

diff --git a/api/src/main/java/ai/djl/util/ClassLoaderUtils.java b/api/src/main/java/ai/djl/util/ClassLoaderUtils.java
@@ -82,7 +82,11 @@ public static <T> T findImplementation(Path path, Class<T> type, String classNam
                             (PrivilegedAction<ClassLoader>)
                                     () -> new URLClassLoader(urls, contextCl));
             if (className != null && !className.isEmpty()) {
-                return initClass(cl, type, className);
+                T impl = initClass(cl, type, className);
+                if (impl == null) {
+                    logger.warn("Failed to load class: {}", className);
+                }
+                return impl;
             }
 
             T implemented = scanDirectory(cl, type, classesDir);