Update code for Python 3.11 compatibility.

PiperOrigin-RevId: 566973675

Author: yilei

neural_structured_learning/tools/build_graph_test.py

@@ -87,7 +87,7 @@ def _create_graph_file(self):
     return self.create_tempfile('graph.tsv').full_path
 
   def _num_file_lines(self, graph_path):
-    with open(graph_path, 'rU') as f:
+    with open(graph_path, 'r') as f:
       return sum(1 for _ in f)
 
   def testBuildGraphInvalidLshBitsValue(self):

neural_structured_learning/tools/graph_utils.py

@@ -124,7 +124,7 @@ def read_tsv_graph(filename):
   logging.info('Reading graph file: %s...', filename)
   graph = {}
   edge_cnt = 0
-  with open(filename, 'rU') as f:
+  with open(filename, 'r') as f:
     for tsv_line in f:
       edge = tsv_line.rstrip('\n').split('\t')
       if add_edge(graph, edge): edge_cnt += 1