Replace CHECK with assert. (jax-ml#2569)

khcs · Apr 1, 2020 · 1d09b6b · 1d09b6b
1 parent ca23be6
commit 1d09b6b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jaxlib/pytree.cc b/jaxlib/pytree.cc
@@ -321,7 +321,7 @@ void PyTreeDef::FlattenHelper(py::handle handle, py::list* leaves,
       FlattenHelper(entry, leaves, tree);
     }
   } else {
-    CHECK(node.kind == Kind::kLeaf);
+    assert(node.kind == Kind::kLeaf);
     leaves->append(py::reinterpret_borrow<py::object>(handle));
   }
   node.num_nodes = tree->traversal_.size() - start_num_nodes + 1;