incorporate more comments

Author: makslevental

mlir/lib/Bindings/Python/MainModule.cpp

@@ -12,8 +12,6 @@
 #include "IRModule.h"
 #include "Pass.h"
 
-#include <tuple>
-
 namespace py = pybind11;
 using namespace mlir;
 using namespace py::literals;

mlir/test/python/dialects/python_test.py

@@ -428,7 +428,7 @@ def __str__(self):
  d = tensor.EmptyOp([1, 2, 3], IntegerType.get_signless(5)).result
  # CHECK: Value(%{{.*}} = tensor.empty() : tensor<1x2x3xi5>)
  print(d)
- # CHECK: <importlib._bootstrap.TestTensorValue object at
+ # CHECK: TestTensorValue
  print(repr(d))
 
 

mlir/test/python/lib/PythonTestModule.cpp

@@ -94,6 +94,9 @@ PYBIND11_MODULE(_mlirPythonTest, m) {
  py::object capsule = mlirApiObjectToCapsule(valueObj);
  MlirValue v = mlirPythonCapsuleToValue(capsule.ptr());
  MlirType t = mlirValueGetType(v);
+ // This is hyper-specific in order to exercise/test registering a
+ // value caster from cpp (but only for a single test case; see
+ // testTensorValue python_test.py).
  if (mlirShapedTypeHasStaticShape(t) &&
  mlirShapedTypeGetDimSize(t, 0) == 1 &&
  mlirShapedTypeGetDimSize(t, 1) == 2 &&