setup.py (prophet-0.7) | : | setup.py (prophet-1.0) | ||
---|---|---|---|---|
skipping to change at line 27 | skipping to change at line 27 | |||
from setuptools.command.build_py import build_py | from setuptools.command.build_py import build_py | |||
from setuptools.command.develop import develop | from setuptools.command.develop import develop | |||
from setuptools.command.test import test as test_command | from setuptools.command.test import test as test_command | |||
from typing import List | from typing import List | |||
PLATFORM = 'unix' | PLATFORM = 'unix' | |||
if platform.platform().startswith('Win'): | if platform.platform().startswith('Win'): | |||
PLATFORM = 'win' | PLATFORM = 'win' | |||
MODEL_DIR = os.path.join('stan', PLATFORM) | MODEL_DIR = os.path.join('stan', PLATFORM) | |||
MODEL_TARGET_DIR = os.path.join('fbprophet', 'stan_model') | MODEL_TARGET_DIR = os.path.join('prophet', 'stan_model') | |||
def get_backends_from_env() -> List[str]: | def get_backends_from_env() -> List[str]: | |||
from fbprophet.models import StanBackendEnum | from prophet.models import StanBackendEnum | |||
return os.environ.get("STAN_BACKEND", StanBackendEnum.PYSTAN.name).split("," ) | return os.environ.get("STAN_BACKEND", StanBackendEnum.PYSTAN.name).split("," ) | |||
def build_models(target_dir): | def build_models(target_dir): | |||
from fbprophet.models import StanBackendEnum | from prophet.models import StanBackendEnum | |||
for backend in get_backends_from_env(): | for backend in get_backends_from_env(): | |||
StanBackendEnum.get_backend_class(backend).build_model(target_dir, MODEL _DIR) | StanBackendEnum.get_backend_class(backend).build_model(target_dir, MODEL _DIR) | |||
class BuildPyCommand(build_py): | class BuildPyCommand(build_py): | |||
"""Custom build command to pre-compile Stan models.""" | """Custom build command to pre-compile Stan models.""" | |||
def run(self): | def run(self): | |||
if not self.dry_run: | if not self.dry_run: | |||
target_dir = os.path.join(self.build_lib, MODEL_TARGET_DIR) | target_dir = os.path.join(self.build_lib, MODEL_TARGET_DIR) | |||
self.mkpath(target_dir) | self.mkpath(target_dir) | |||
skipping to change at line 68 | skipping to change at line 68 | |||
develop.run(self) | develop.run(self) | |||
class TestCommand(test_command): | class TestCommand(test_command): | |||
user_options = [ | user_options = [ | |||
('test-module=', 'm', "Run 'test_suite' in specified module"), | ('test-module=', 'm', "Run 'test_suite' in specified module"), | |||
('test-suite=', 's', | ('test-suite=', 's', | |||
"Run single test, case or suite (e.g. 'module.test_suite')"), | "Run single test, case or suite (e.g. 'module.test_suite')"), | |||
('test-runner=', 'r', "Test runner to use"), | ('test-runner=', 'r', "Test runner to use"), | |||
('test-slow', 'w', "Test slow suites (default off)"), | ('test-slow', 'w', "Test slow suites (default off)"), | |||
] | ] | |||
test_slow = None | ||||
def initialize_options(self): | def initialize_options(self): | |||
super(TestCommand, self).initialize_options() | super(TestCommand, self).initialize_options() | |||
self.test_slow = False | self.test_slow = False | |||
def finalize_options(self): | def finalize_options(self): | |||
super(TestCommand, self).finalize_options() | super(TestCommand, self).finalize_options() | |||
if self.test_slow is None: | if self.test_slow is None: | |||
self.test_slow = getattr(self.distribution, 'test_slow', False) | self.test_slow = getattr(self.distribution, 'test_slow', False) | |||
skipping to change at line 118 | skipping to change at line 119 | |||
sys.modules.update(old_modules) | sys.modules.update(old_modules) | |||
working_set.__init__() | working_set.__init__() | |||
with open('README.md', 'r', encoding='utf-8') as f: | with open('README.md', 'r', encoding='utf-8') as f: | |||
long_description = f.read() | long_description = f.read() | |||
with open('requirements.txt', 'r') as f: | with open('requirements.txt', 'r') as f: | |||
install_requires = f.read().splitlines() | install_requires = f.read().splitlines() | |||
setup( | setup( | |||
name='fbprophet', | name='prophet', | |||
version='0.7.1', | version='1.0.1', | |||
description='Automatic Forecasting Procedure', | description='Automatic Forecasting Procedure', | |||
url='https://facebook.github.io/prophet/', | url='https://facebook.github.io/prophet/', | |||
author='Sean J. Taylor <sjtz@pm.me>, Ben Letham <bletham@fb.com>', | author='Sean J. Taylor <sjtz@pm.me>, Ben Letham <bletham@fb.com>', | |||
author_email='sjtz@pm.me', | author_email='sjtz@pm.me', | |||
license='MIT', | license='MIT', | |||
packages=find_packages(), | packages=find_packages(), | |||
setup_requires=[ | setup_requires=[ | |||
], | ], | |||
install_requires=install_requires, | install_requires=install_requires, | |||
python_requires='>=3', | python_requires='>=3', | |||
zip_safe=False, | zip_safe=False, | |||
include_package_data=True, | include_package_data=True, | |||
cmdclass={ | cmdclass={ | |||
'build_py': BuildPyCommand, | 'build_py': BuildPyCommand, | |||
'develop': DevelopCommand, | 'develop': DevelopCommand, | |||
'test': TestCommand, | 'test': TestCommand, | |||
}, | }, | |||
test_suite='fbprophet.tests', | test_suite='prophet.tests', | |||
classifiers=[ | classifiers=[ | |||
'Programming Language :: Python', | 'Programming Language :: Python', | |||
'Programming Language :: Python :: 3', | 'Programming Language :: Python :: 3', | |||
'Programming Language :: Python :: 3.7', | 'Programming Language :: Python :: 3.7', | |||
], | ], | |||
long_description=long_description, | long_description=long_description, | |||
long_description_content_type='text/markdown', | long_description_content_type='text/markdown', | |||
) | ) | |||
End of changes. 6 change blocks. | ||||
6 lines changed or deleted | 7 lines changed or added |