setup.py

#!/usr/bin/env python3
#
# Copyright (c) Facebook, Inc. and its affiliates.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.

import distutils.util

import setuptools

with open("../VERSION") as f:
    version = f.read().strip()
with open("requirements.txt") as f:
    requirements = [ln.split("#")[0].rstrip() for ln in f.readlines()]
with open("../tests/requirements.txt") as f:
    requirements += [ln.split("#")[0].rstrip() for ln in f.readlines()]

setuptools.setup(
    name="compiler_gym_examples",
    version=version,
    description="Example code for CompilerGym",
    author="Facebook AI Research",
    url="https://github.com/facebookresearch/CompilerGym",
    license="MIT",
    install_requires=requirements,
    packages=[
        "llvm_autotuning",
        "llvm_autotuning.autotuners",
        "llvm_rl",
        "llvm_rl.model",
    ],
    package_data={
        "llvm_autotuning": [
            "config/*.yaml",
            "config/**/*.yaml",
        ],
        "llvm_rl": [
            "config/*.yaml",
            "config/**/*.yaml",
        ],
    },
    python_requires=">=3.8",
    platforms=[distutils.util.get_platform()],
    zip_safe=False,
)