fix ci

apache · Jan 27, 2021 · c484dd8 · c484dd8
1 parent 2889f79
commit c484dd8
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/printer/relay_text_printer.cc b/src/printer/relay_text_printer.cc
@@ -827,9 +827,11 @@ std::vector<Doc> RelayTextPrinter::PrintCallAttrs(const Attrs& attrs, const Expr
   } else {
     AttrPrinter printer(&docs, this);
     const_cast<BaseAttrsNode*>(attrs.operator->())->VisitNonDefaultAttrs(&printer);
-    // print call attr type key to restore expr for relay parser
-    std::string s = std::string(attrs->GetTypeKey());
-    printer.Visit("attrs_type_key", &s);
+    if (!op_node) {
+      // print call attr type key to restore expr for relay parser
+      std::string s = std::string(attrs->GetTypeKey());
+      printer.Visit("attrs_type_key", &s);
+    }
     return docs;
   }
 }