pytsa/setup.py at master · bkmarzouk/pytsa · GitHub

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
import os
import subprocess
from setuptools import setup

sym_cache = os.path.join(os.path.dirname(__file__), "pytsa", "sym_cache")
if not os.path.exists(sym_cache):
    os.makedirs(sym_cache)

for sub_dir in ["model", "fmet", "pot", "covd"]:
    sub_dir = os.path.join(sym_cache, sub_dir)

    if not os.path.exists(sub_dir):
        os.makedirs(sub_dir)

default_samplers_cache = os.path.join(os.path.dirname(__file__), "samplers")

if not os.path.exists(default_samplers_cache):
    os.makedirs(default_samplers_cache)

setup(
    name="pytsa",
    version="0.1.0",
    author="Kareem Marzouk",
    author_email="bkmarzouk@gmail.com",
    packages=["pytsa"],
    description="This package contains the PyTransport source, "
    "interfaced with sampling technology.",
    url="https://github.com/bkmarzouk/pytsa",
    install_requires=[
        "scipy>=1.6.1",
        "sympy>=1.6.1",
        "numpy>=1.20.1",
        "dill>=0.3.3",
        "mpi4py>=3.0.3",
        "schwimmbad>=0.3.2",
        "pandas>=1.1.1",
    ],
    python_requires=">=3.8, <4",
)

print("attempting installation of dquad_2sphere.py example model file...")
model_path = os.path.abspath(
    os.path.join(
        os.path.dirname(__file__), "example_models", "dquad_2sphere.py"
    )
)
subprocess.run(["python", model_path], timeout=120)

try:
    from pytsa.models import dquad_2sphere

    print("model file successfully installed!")
    del dquad_2sphere

except ImportError:
    print("failed to install model file.")