Migrate black linter on custom_lint pipeline

1 files changed, 25 insertions, 0 deletions

diff --git a/library/cpp/pybind/init.h b/library/cpp/pybind/init.h
new file mode 100644
index 00000000000..58874574ed6
--- /dev/null
+++ b/library/cpp/pybind/init.h
@@ -0,0 +1,25 @@
+#pragma once
+
+#define PY_SSIZE_T_CLEAN
+#include <Python.h>
+
+#include "ptr.h"
+
+namespace NPyBind {
+#if PY_MAJOR_VERSION >= 3
+
+#define PYBIND_MODINIT(name) PyMODINIT_FUNC PyInit_##name()
+
+    inline PyObject* ModInitReturn(TPyObjectPtr&& modptr) {
+        return modptr.Release();
+    }
+
+#else
+
+#define PYBIND_MODINIT(name) PyMODINIT_FUNC init##name()
+
+    inline void ModInitReturn(TPyObjectPtr&&) {
+    }
+
+#endif
+}