From 07cd40297797d6f8a2e3835e6c65636e11d0e7ae Mon Sep 17 00:00:00 2001
From: Frank Liu <frankfliu2000@gmail.com>
Date: Thu, 4 May 2023 15:45:32 -0700
Subject: [PATCH] [api] Allows subclass access member variable of Predictor

---
 api/src/main/java/ai/djl/inference/Predictor.java | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/api/src/main/java/ai/djl/inference/Predictor.java b/api/src/main/java/ai/djl/inference/Predictor.java
index 3d512173698..da1cda49a3f 100644
--- a/api/src/main/java/ai/djl/inference/Predictor.java
+++ b/api/src/main/java/ai/djl/inference/Predictor.java
@@ -84,11 +84,12 @@
 public class Predictor<I, O> implements AutoCloseable {
 
     private static final Logger logger = LoggerFactory.getLogger(Predictor.class);
-    private Translator<I, O> translator;
-    private long timestamp;
 
-    private boolean prepared;
-    private Model model;
+    protected Translator<I, O> translator;
+    protected long timestamp;
+
+    protected boolean prepared;
+    protected Model model;
     protected NDManager manager;
     protected Metrics metrics;
     protected Block block;
@@ -357,12 +358,12 @@ protected void finalize() throws Throwable {
         super.finalize();
     }
 
-    private class PredictorContext implements TranslatorContext {
+    protected class PredictorContext implements TranslatorContext {
 
         private NDManager ctxManager;
         private Map<String, Object> attachments;
 
-        PredictorContext() {
+        protected PredictorContext() {
             ctxManager = manager.newSubManager();
             ctxManager.setName("predictor ctx");
             attachments = new ConcurrentHashMap<>();