setup.py

from setuptools import setup, find_packages
import os
from glob import glob

package_name = 'general_screw_axis_prediction'

setup(
    name=package_name,
    version='0.1.0',
    packages=find_packages() + ['scripts'],
    data_files=[
        ('share/ament_index/resource_index/packages', ['resource/' + package_name]),
        ('share/' + package_name, ['package.xml']),
        (os.path.join('share', package_name, 'launch'), glob('launch/*.launch.py')),
        (os.path.join('lib', package_name), glob('scripts/*.sh')),
        # NEW: install config folder so ground_truth_screw.json is findable
        (os.path.join('share', package_name, 'config', 'objects'),
            glob('config/objects/*.yaml')),
    ],
    maintainer='Daniel Adebi',
    maintainer_email='ikadebi@utexas.edu',
    description='Screw axis prediction from RGB + SAM2 masks using DINOv2',
    license='Apache-2.0',
    install_requires=['setuptools'],
    entry_points={
        'console_scripts': [
            'lcamp_service_node          = scripts.lcamp_service_node:main',
            'save_gt_screws              = scripts.save_gt_screws:main',
            'gt_screw_server             = scripts.gt_screw_server:main',
        ],
    },
)