Fix pylint

Signed-off-by: Partho Sarthi <[email protected]>
NVIDIA · Dec 12, 2023 · ce3ccf0 · ce3ccf0
1 parent bdc65e2
commit ce3ccf0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/user_tools/src/spark_rapids_pytools/common/cluster_inference.py b/user_tools/src/spark_rapids_pytools/common/cluster_inference.py
@@ -37,7 +37,7 @@ class ClusterInference:
     logger: Logger = field(default=ToolLogging.get_and_setup_logger('rapids.tools.cluster_inference'), init=False)
 
     @staticmethod
-    def _load_event_logs(eventlog_arg) -> list[JSONPropertiesContainer]:
+    def _load_event_logs(eventlog_arg) -> list:
         """
         Read the event log file and return a list of event log properties containers.
         """
@@ -48,7 +48,7 @@ def _load_event_logs(eventlog_arg) -> list[JSONPropertiesContainer]:
             event_logs = [JSONPropertiesContainer(line, file_load=False) for line in file]
         return event_logs
 
-    def get_cluster_info(self, eventlog_props: list[JSONPropertiesContainer]):
+    def get_cluster_info(self, eventlog_props: list):
         """
         Process event logs and extract information about drivers and executors.
         """