polish code.

hbwx24 · hbwx24 · commit ab9fdeed94b1 · 2021-08-26T05:59:39.000Z
diff --git a/python/paddle/fluid/variable_index.py b/python/paddle/fluid/variable_index.py
@@ -129,8 +129,7 @@ def get_item(self, tensor):
 
     def set_item(self, tensor_origin, value):
 
-        if not isinstance(value,
-                          (paddle.fluid.core.VarBase, paddle.fluid.Variable)):
+        if not isinstance(value, paddle.fluid.Variable):
             value = paddle.assign(value)
         tensor_type = None
 
