- lints

Author: mbs-octoml

src/printer/doc.cc

@@ -52,9 +52,7 @@ TVM_REGISTER_OBJECT_TYPE(DocTextNode);
 
 class DocText : public DocAtom {
  public:
-  explicit DocText(std::string str) {
-    data_ = runtime::make_object<DocTextNode>(str);
-  }
+  explicit DocText(std::string str) { data_ = runtime::make_object<DocTextNode>(str); }
 
   TVM_DEFINE_OBJECT_REF_METHODS(DocText, DocAtom, DocTextNode);
 };

src/support/scalars.cc

@@ -206,4 +206,4 @@ std::pair<FloatImm, bool> ValueToFloatImm(double value, int width) {
 }
 
 }  // namespace support
-}  // namespace tvm
+}  // namespace tvm

src/support/scalars.h

@@ -25,6 +25,9 @@
 #ifndef TVM_SUPPORT_SCALARS_H_
 #define TVM_SUPPORT_SCALARS_H_
 
+#include <string>
+#include <utility>
+
 #include "tvm/ir/expr.h"
 #include "tvm/relay/expr.h"
 #include "tvm/runtime/ndarray.h"
@@ -60,4 +63,4 @@ std::pair<FloatImm, bool> ValueToFloatImm(double value, int width);
 }  // namespace support
 }  // namespace tvm
 
-#endif  // TVM_SUPPORT_SCALARS_H_
+#endif  // TVM_SUPPORT_SCALARS_H_

tests/python/relay/test_ir_parser.py

@@ -189,6 +189,7 @@ def test_int_literal():
         parse_text("2147483648i32")
         parse_text("32768i16")
 
+
 def test_float_literal():
     assert get_scalar(parse_text("1.0f")) == 1.0
     assert isclose(get_scalar(parse_text("1.56667f")), 1.56667)

tests/python/relay/test_ir_text_printer.py

@@ -46,9 +46,11 @@ def show(text):
         print("---------------------------")
         print(text)
 
+
 def assert_prints_as(expr, str):
     assert astext(expr) == SEMVER + str
 
+
 def test_scalars():
     assert_prints_as(relay.const(42, "int16"), "42i16")
     assert_prints_as(relay.const(42, "int32"), "42")
@@ -57,6 +59,7 @@ def test_scalars():
     assert_prints_as(relay.const(3.0, "float32"), "3f")
     assert_prints_as(relay.const(3.0, "float64"), "3f64")
 
+
 def test_large_graph():
     x = relay.var("x", shape=(3, 2))
     y = relay.var("y")