Spaces:
Running
Running
File size: 4,878 Bytes
1721aea |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 |
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,
}
) |