/
pypi.py
37 lines (32 loc) · 1022 Bytes
/
pypi.py
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
PYPI = {'tsa':'statsmodels',
'fbprophet':'prophet',
'pmd':'pmdarima',
'rvr':'river',
'nprophet':'neuralprophet',
'dlm':'pydlm',
'divine':'divinity',
'orbit':'orbit-ml',
'bats':'tbats',
'glu':'gluonts',
'flux':'pyflux',
'sk':'sktime',
'smdk':'simdkalman',
'gk':'greykite',
'tcn':'keras-tcn',
'darts':'darts',
'kts':'kats',
'ats':'auto_ts',
'suc':'successor',
'mrln':'salesforce-merlion',
'pycrt':'pycaret',
'ik':'scikit-learn'}
def pypi_from_name(name):
stem = name.split('_')[0]
short_name = PYPI.get(stem)
stub = 'https://pypi.org/project/'
return stub+short_name if short_name else stub+'timemachines'
if __name__=='__main__':
from timemachines.skaters.localskaters import LOCAL_SKATERS
from pprint import pprint
pprint([(sk.__name__,pypi_from_name(sk.__name__)) for sk in LOCAL_SKATERS])
print(len(LOCAL_SKATERS))