Fix wheel build

Author: zkytony

CHANGELOG.rst

@@ -10,6 +10,7 @@ Version 1.3.3 (07/25/2023)
 * Add :code:`__init__` signature for Environment in comments to be visible in docs
 * Fix :code:`s -> sp` in :py:mod:`~pomdp_py.algorithms.value_iteration.ValueIteration` (`pomdp-py#20 <https://github.com/h2r/pomdp-py/issues/20>`_)
 * Allow updating rollout policy for :py:mod:`~pomdp_py.algorithms.po_uct.POUCT` and :py:mod:`~pomdp_py.algorithms.pomcp.POMCP`
+* Fix in :code:`setup.py` so that wheel builds properly.
 * Minor bug fixes and documentation.
 
 

setup.py

@@ -52,7 +52,7 @@ def build_extensions(pkg_name, major_submodules):
       ext_modules=cythonize(extensions,
                             build_dir="build",
                             compiler_directives={'language_level' : "3"}),
-      package_data={"pomdp_py": ["*.pxd", "*.pyx"],
-                    "pomdp_problems": ["*.pxd", "*.pyx"]},
+      package_data={"pomdp_py": ["*.pxd", "*.pyx", "*.so", "*.c"],
+                    "pomdp_problems": ["*.pxd", "*.pyx", "*.so", "*.c"]},
       zip_safe=False
 )