From 814f6c5553d9b641bcb1b2a10a4d6a7fcd850eb5 Mon Sep 17 00:00:00 2001
From: Graph Learning Team <no-reply@google.com>
Date: Tue, 14 Mar 2023 11:50:46 -0700
Subject: [PATCH] Fix or ignore some pytype errors.

PiperOrigin-RevId: 516590021
---
 tensorflow_gnn/graph/graph_piece.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow_gnn/graph/graph_piece.py b/tensorflow_gnn/graph/graph_piece.py
index a158be0f..e4709870 100644
--- a/tensorflow_gnn/graph/graph_piece.py
+++ b/tensorflow_gnn/graph/graph_piece.py
@@ -281,7 +281,7 @@ def update_fn(value):
                                       new_spec._metadata)
       return value
 
-    return self.__class__(
+    return self.__class__(  # pytype: disable=not-instantiable
         tf.nest.map_structure(update_fn, self._data), new_spec)
 
   @property