setup.py

from setuptools import setup, find_packages
from os.path import join, dirname

all_packages = []
for pkg in find_packages():
    if "demos" in pkg:
        pkg = "ocelot." +  pkg
    all_packages.append(pkg)


setup(
    name='ocelot',
    version='19.06.0',
    description='Accelerator, radiation and x-ray optics simulation framework',
    author='ocelot-collab',
    author_email='tomin.sergey@gmail.com',
    url='https://github.com/ocelot-collab/ocelot',
    packages=all_packages,
    package_dir={'ocelot.demos': 'demos'},  ## install examples along with the rest of the source
    install_requires=[
        'numpy', 'scipy', 'matplotlib'
    ],
    package_data={'ocelot.optics': ['data/*.dat']},
    # long_description=open(join(dirname(__file__), 'README.txt')).read(),
)