(*args, **kwargs)
| 103 | method = getattr(NeutronPartitioner, "partition") # noqa B009 |
| 104 | |
| 105 | def wrapper(*args, **kwargs): |
| 106 | result = method(*args, **kwargs) |
| 107 | visualize_with_clusters( |
| 108 | result.tagged_exported_program, |
| 109 | os.path.join(test_dir, test_name + "_partitioned.json"), |
| 110 | False, |
| 111 | ) |
| 112 | return result |
| 113 | |
| 114 | wrapped = functools.update_wrapper(wrapper, method) |
| 115 | mocker.patch.object( |
nothing calls this directly
no test coverage detected