From b72e893d1d8e198f8f1674c393d27e07423c377e Mon Sep 17 00:00:00 2001
From: Mircea Trofin <mtrofin@google.com>
Date: Fri, 3 Feb 2023 18:03:39 -0800
Subject: [PATCH] [mlgo] Fix type annotation in log_reader, for older python3
 versions

---
 llvm/lib/Analysis/models/log_reader.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/llvm/lib/Analysis/models/log_reader.py b/llvm/lib/Analysis/models/log_reader.py
index 0fe090c..7278b1f 100644
--- a/llvm/lib/Analysis/models/log_reader.py
+++ b/llvm/lib/Analysis/models/log_reader.py
@@ -8,7 +8,7 @@ import io
 import json
 import math
 import sys
-from typing import Optional
+from typing import List, Optional
 
 _element_types = {
     'float': ctypes.c_float,
@@ -28,7 +28,7 @@ _element_types = {
 class TensorSpec:
   name: str
   port: int
-  shape: list[int]
+  shape: List[int]
   element_type: type
 
   @staticmethod
@@ -89,7 +89,7 @@ def read_header(f: io.BufferedReader):
 def read_one_observation(context: Optional[str],
                          event_str: str,
                          f: io.BufferedReader,
-                         tensor_specs: list[TensorSpec],
+                         tensor_specs: List[TensorSpec],
                          score_spec: Optional[TensorSpec]):
   event = json.loads(event_str)
   if 'context' in event:
-- 
2.7.4