Class PrintOptionsContext

python/src/print.cpp:16–28

using namespace nb::literals;

struct PrintOptionsContext {
  mx::PrintOptions old_options;
  mx::PrintOptions new_options;
  PrintOptionsContext(mx::PrintOptions p) : new_options(p) {}
  PrintOptionsContext& enter() {
    old_options = mx::get_global_formatter().format_options;
    mx::set_printoptions(new_options);
    return *this;
  }
  void exit(nb::args) {
    mx::set_printoptions(old_options);
  }
};

void init_print(nb::module_& m) {
  // Set Python print formatting options