From 728d64dac153fe3880a7cbf48439490d2b30e1b0 Mon Sep 17 00:00:00 2001
From: Qing Lan <qingla@amazon.com>
Date: Thu, 2 May 2024 18:27:39 -0700
Subject: [PATCH] remove unused field

---
 .../djl_python/tests/test_rolling_batch.py    | 36 +------------------
 1 file changed, 1 insertion(+), 35 deletions(-)

diff --git a/engines/python/setup/djl_python/tests/test_rolling_batch.py b/engines/python/setup/djl_python/tests/test_rolling_batch.py
index bdea028286..91651fa5a9 100644
--- a/engines/python/setup/djl_python/tests/test_rolling_batch.py
+++ b/engines/python/setup/djl_python/tests/test_rolling_batch.py
@@ -1,10 +1,8 @@
 import json
-import os
 import unittest
 
 import djl_python.rolling_batch.rolling_batch
-from djl_python.rolling_batch.rolling_batch import Request, Token, _json_output_formatter, _jsonlines_output_formatter, \
-    RollingBatch
+from djl_python.rolling_batch.rolling_batch import Request, Token, _json_output_formatter, _jsonlines_output_formatter
 
 
 class TestRollingBatch(unittest.TestCase):
@@ -267,22 +265,6 @@ def custom_fmt(token: Token, first_token: bool, last_token: bool,
                 result["finish_reason"] = details["finish_reason"]
             return json.dumps(result) + "\n"
 
-        class CustomRB(RollingBatch):
-
-            def preprocess_requests(self, requests):
-                pass
-
-            def postprocess_results(self):
-                pass
-
-            def inference(self, input_data, parameters):
-                pass
-
-            def get_tokenizer(self):
-                pass
-
-        rb = CustomRB()
-
         req = Request(132,
                       "This is a wonderful day",
                       parameters={
@@ -321,22 +303,6 @@ def custom_fmt(token: Token, first_token: bool, last_token: bool,
             result = details
             return json.dumps(result) + "\n"
 
-        class CustomRB(RollingBatch):
-
-            def preprocess_requests(self, requests):
-                pass
-
-            def postprocess_results(self):
-                pass
-
-            def inference(self, input_data, parameters):
-                pass
-
-            def get_tokenizer(self):
-                pass
-
-        rb = CustomRB()
-
         req = Request(132,
                       "This is a wonderful day",
                       parameters={