Spaces:
Running
Running
| import os | |
| import subprocess | |
| import sys | |
| from setuptools import setup, find_packages | |
| from setuptools.command.install import install | |
| # --- Conda Environment and Dependency Installation --- | |
| def get_conda_path(): | |
| """Tries to find the path to the conda executable.""" | |
| # Common locations for conda executable | |
| # 1. In the system's PATH | |
| try: | |
| conda_path = subprocess.check_output("which conda", shell=True).strip().decode('utf-8') | |
| if conda_path: return conda_path | |
| except subprocess.CalledProcessError: | |
| pass # Not in PATH | |
| # 2. Common installation directories | |
| possible_paths = [ | |
| os.path.expanduser("~/anaconda3/bin/conda"), | |
| os.path.expanduser("~/miniconda3/bin/conda"), | |
| "/opt/anaconda3/bin/conda", | |
| "/opt/miniconda3/bin/conda", | |
| ] | |
| for path in possible_paths: | |
| if os.path.exists(path): | |
| return path | |
| return None | |
| def conda_env_exists(env_name): | |
| """Check if a conda environment with the given name already exists.""" | |
| try: | |
| envs = subprocess.check_output("conda env list", shell=True).decode('utf-8') | |
| return any(line.startswith(env_name + ' ') for line in envs.splitlines()) | |
| except (subprocess.CalledProcessError, FileNotFoundError): | |
| return False | |
| class CondaInstallCommand(install): | |
| """Custom command to create Conda env and install dependencies before package installation.""" | |
| description = "Create Conda environment and install dependencies, then install the package." | |
| def run(self): | |
| env_name = "causal-agent" | |
| conda_path = get_conda_path() | |
| if conda_path: | |
| print(f"--- Found Conda at: {conda_path} ---") | |
| if not conda_env_exists(env_name): | |
| print(f"--- Creating Conda environment: {env_name} ---") | |
| try: | |
| # Create the environment with a specific Python version | |
| subprocess.check_call(f"{conda_path} create -n {env_name} python=3.10 --yes", shell=True) | |
| except subprocess.CalledProcessError as e: | |
| print(f"Error creating conda environment: {e}", file=sys.stderr) | |
| sys.exit(1) | |
| else: | |
| print(f"--- Conda environment '{env_name}' already exists. Skipping creation. ---") | |
| print(f"--- Installing dependencies from requirement.txt into '{env_name}' ---") | |
| try: | |
| # Command to run pip install within the conda environment | |
| pip_install_cmd = f"{conda_path} run -n {env_name} pip install -r requirement.txt" | |
| subprocess.check_call(pip_install_cmd, shell=True) | |
| print("--- Dependencies installed successfully. ---") | |
| except subprocess.CalledProcessError as e: | |
| print(f"Error installing dependencies: {e}", file=sys.stderr) | |
| sys.exit(1) | |
| else: | |
| print("--- Conda not found. Skipping environment creation. ---") | |
| print("--- Please ensure you have created an environment and installed dependencies manually. ---") | |
| # Proceed with the standard installation | |
| super().run() | |
| # --- Standard Setup Configuration --- | |
| # Read the contents of your requirements file | |
| try: | |
| with open('requirement.txt') as f: | |
| requirements = f.read().splitlines() | |
| except FileNotFoundError: | |
| print("requirement.txt not found. Please ensure it is in the root directory.", file=sys.stderr) | |
| requirements = [] | |
| # Read README for long description | |
| try: | |
| with open('README.md', encoding='utf-8') as f: | |
| long_description = f.read() | |
| except FileNotFoundError: | |
| long_description = 'A library for automated causal inference.' | |
| setup( | |
| name='auto_causal', | |
| version='0.1.0', | |
| author='Vishal Verma', | |
| author_email='vishal.verma@andrew.cmu.edu', | |
| description='A library for automated causal inference', | |
| long_description=long_description, | |
| long_description_content_type='text/markdown', | |
| url='https://github.com/causalNLP/causal-agent', | |
| packages=find_packages(exclude=['tests', 'tests.*']), | |
| install_requires=requirements, | |
| classifiers=[ | |
| 'Development Status :: 3 - Alpha', | |
| 'Intended Audience :: Developers', | |
| 'Intended Audience :: Science/Research', | |
| 'License :: OSI Approved :: MIT License', | |
| 'Programming Language :: Python :: 3', | |
| 'Programming Language :: Python :: 3.8', | |
| 'Programming Language :: Python :: 3.9', | |
| 'Programming Language :: Python :: 3.10', | |
| 'Operating System :: OS Independent', | |
| 'Topic :: Scientific/Engineering :: Artificial Intelligence', | |
| 'Topic :: Scientific/Engineering :: Information Analysis', | |
| ], | |
| python_requires='>=3.8', | |
| include_package_data=True, | |
| zip_safe=False, | |
| cmdclass={ | |
| 'install': CondaInstallCommand, | |
| } | |
| ) |