merge

assapin · Nov 5, 2023 · 32f5cb5 · 32f5cb5
1 parent 21e3316
commit 32f5cb5
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 4 deletions.
diff --git a/execution/py_tester_bindings/make_python_bindings.sh b/execution/py_tester_bindings/make_python_bindings.sh
@@ -10,3 +10,9 @@ bazel build //execution/py_tester_bindings:py_tester_extention.so --
 cp bazel-bin/execution/py_tester_bindings/py_tester_extention.so execution/py_tester_bindings/code_contests_tester/
 /usr/local/bin/pip wheel execution/py_tester_bindings 	--no-deps -w wheelhouse/
 
+
+pybind11-stubgen py_tester_extention
+
+python setup.py sdist bdist_wheel
+pip install dist/code_contests_tester-0.1-py3-none-any.whl --force-reinstall
+
diff --git a/execution/py_tester_bindings/py_tester_sandboxer_bindings.cc b/execution/py_tester_bindings/py_tester_sandboxer_bindings.cc
@@ -54,6 +54,8 @@ PYBIND11_MODULE(py_tester_extention, m) {
     .def(py::init<>())
     .def_readwrite("num_threads", &TestOptions::num_threads)
     .def_readwrite("stop_on_first_failure", &TestOptions::stop_on_first_failure)
+    .def_readwrite("max_execution_duration", &TestOptions::max_execution_duration)
+
     // ... other fields ...
     ;
 
@@ -102,7 +104,5 @@ PYBIND11_MODULE(py_tester_extention, m) {
 
         return result.value();
     });
-    // If there are public methods in the `TesterSandboxer` or `PyTesterSandboxer` 
-    // classes that you want to expose, you can continue the bindings in a similar manner.
 }
 
diff --git a/execution/py_tester_bindings/requirements.txt b/execution/py_tester_bindings/requirements.txt
@@ -1,2 +1,3 @@
 pybind11==2.11.1
 wheel
+pybind11-stubgen==2.3.6
diff --git a/execution/py_tester_bindings/setup.py b/execution/py_tester_bindings/setup.py
@@ -10,10 +10,11 @@
     setup_requires=[
         'setuptools',
         'wheel',
-        'pybind11==2.11.1'
+        'pybind11==2.11.1',
+        'pybind11-stubgen==2.3.6'
     ],
     package_data={
-        'code_contests_tester': [f'py_tester_extention.so'],
+        'code_contests_tester': [f'py_tester_extention.so', 'py_tester_extention.pyi'],
     },
 )