IBM_AI/vi-opton/opton_pred_v2/config.py

_model_version = {
    'ZZSB-CLS-Task':            '2020080101',

    'ZZSB-OD-alphaF':           '2020092601',

    'ZZSB-OD-B':                '2020060801',
    'ZZSB-OD-P':                '2020020301',
    'ZZSB-OD-F':                '2020020301',
    'ZZSB-OD-seg_B_M':          '2020020301',
    'ZZSB-OD-PS_C':             '2020030101',

    'ZZSB-OD-M_B':              '2020062201',

    'ZZSB-OD-deltaF':           '2020062301',
    'ZZSB-OD-C':                '2019110501',
    
    'BZPJ-OD-ZT_H-G_XZ_BCF':    '2020112401',
    'BZPJ-OD-ZT_H-G_XZ_DE':     '2020112401',
    
    'BZPJ-OD-ZT_H-LGJ_SL':      '2020060801',

    'BZPJ-OD-ZT_H-GJT_SL':      '2020082601',
    'BZPJ-CLS-ZT_H-GJT_SL':     '202012031445',

    'BZPJ-OD-ZT_QM-LGJ_SL':     '2019123001',

    'BZPJ-OD-ZT_QM-G':          '2020061001',

    'BZPJ-CLS-ZCG_GTG-QH':      '2020080301',
    'BZPJ-OD-ZCG_GTG-WZ':       '2020020301',
    'BZPJ-CLS-ZCG_GTG-DZ':      '2020020302',

    'BZPJ-OD-W_A':              '2020063001',
    'BZPJ-OD-W_B':              '2020070101',

    'BZPJ-OD-YBG-JZW_ABC':      '2020110501',
    'BZPJ-OD-YBG-JZW_DDS':      '20201224',

    'BZPJ-CLS-YBG-JLD':         '20210202',
    'BZPJ-CLS-YGXG-DZ_A':       '20210104',
    'BZPJ-CLS-YGXG-DZ_B':       '2020093001',
    'BZPJ-CLS-YGXG-DZ_C':       '2020101601',
    'BZPJ-CLS-YGXG-DZ_D':       '2020101601',
    'BZPJ-CLS-YGXG-DZ_E':       '2020101601',

    'Ruler-OD':                 '2020121001',

    'BZPJ-OD-COAL':             '20210409',
    'BZPJ-OD-COAL_BIG':         '2021040902',
    'BZPJ-OD-HZ':               '20210414',
    'BZPJ-OD-HZ_DL':            '20210415',
    'BZPJ-OD-KLQ':              '20210426small',
    'BZPJ-CLS-ZZ':              '20210207',

    'BZPJ-CLS-ZZ':              '20210207',
}

_task_models = {
    'Ruler': [
        'Ruler-OD-latest',
        ],

    'ZZSB-CLS-Task': [
        'ZZSB-CLS-Task-latest',
        ],

    'ZZSB-a-alpha_f': [
        'ZZSB-OD-alphaF-latest',
        ],

    'ZZSB-b-p-f': [
        'ZZSB-OD-B-latest',
        'ZZSB-OD-P-latest',
        'ZZSB-OD-F-latest',
        'BZPJ-OD-W_A-latest',
        'BZPJ-OD-W_B-latest',
        'ZZSB-OD-seg_B_M-latest',
        ],

    'ZZSB-f-p': [
        'ZZSB-OD-B-latest',
        'ZZSB-OD-P-latest',
        'ZZSB-OD-F-latest',
        'BZPJ-OD-W_A-latest',
        'BZPJ-OD-W_B-latest',
        'ZZSB-OD-seg_B_M-latest',
        ],

    'ZZSB-p': [
        'ZZSB-OD-F-latest',
        ],

    'ZZSB-p-f': [
        'ZZSB-OD-F-latest',
        'BZPJ-OD-W_A-latest',
        'BZPJ-OD-W_B-latest',
        'ZZSB-OD-seg_B_M-latest',
        ],

    'ZZSB-g_p-ps_c': [
        'ZZSB-OD-PS_C-latest',
        ],

    'ZZSB-m': [
        'ZZSB-OD-M_B-latest',
        ],

    'ZZSB-temps': [
        'ZZSB-OD-deltaF-latest',
        'ZZSB-OD-C-latest',
        ],

    'BZPJ-ZT_H-G_XZ':  [
        'BZPJ-OD-ZT_H-G_XZ_BCF-latest',
        'BZPJ-OD-ZT_H-G_XZ_DE-latest',
        ],

    'BZPJ-ZT_H-LGJ_SL':  [
        'BZPJ-OD-ZT_H-LGJ_SL-latest'
        ],

    'BZPJ-ZT_H-GJT_SL_OD':  [
        'BZPJ-OD-ZT_H-GJT_SL-latest'
        ],
    'BZPJ-ZT_H-GJT_SL_CLS':  [
        'BZPJ-CLS-ZT_H-GJT_SL-latest'
        ],

    'BZPJ-ZT_H-P_SL':  [
        'ZZSB-OD-F-latest',
        ],

    'BZPJ-ZT_QM-LGJ_SL':  [
        'BZPJ-OD-ZT_QM-LGJ_SL-latest'
        ],

    'BZPJ-ZT_QM-P_SL':  [
        'ZZSB-OD-F-latest',
        'BZPJ-OD-ZT_QM-G-latest',
        ],

    'BZPJ-ZT_QM-F_SL':  [
        'ZZSB-OD-F-latest',
        'BZPJ-OD-ZT_QM-G-latest',
        ],

    'BZPJ-ZCG_GTG-QH':  [
        'BZPJ-CLS-ZCG_GTG-QH-latest'
        ],

    'BZPJ-ZCG_GTG-WZ_T':  [
        'BZPJ-OD-ZCG_GTG-WZ-latest'
        ],

    'BZPJ-ZCG_GTG-DZ':  [
        'BZPJ-CLS-ZCG_GTG-DZ-latest'
        ],

    'BZPJ-YGXG-DZ_E':  [
        'ZZSB-OD-F-latest'
        ],

    'BZPJ-YBG-JZW':  [
        'BZPJ-OD-YBG-JZW_ABC-latest',
        'BZPJ-OD-YBG-JZW_DDS-latest'
        ],

    'BZPJ-YBG-JLD_OD':  [
        'ZZSB-OD-F-latest'
        ],
    'BZPJ-YBG-JLD_CLS':  [
        'BZPJ-CLS-YBG-JLD-latest'
        ],
    'BZPJ-YGXG-DZ_A_CLS':  [
        'BZPJ-CLS-YGXG-DZ_A-latest'
        ],
    'BZPJ-YGXG-DZ_B_CLS':  [
        'BZPJ-CLS-YGXG-DZ_B-latest'
        ],
    'BZPJ-YGXG-DZ_C_CLS':  [
        'BZPJ-CLS-YGXG-DZ_C-latest'
        ],
    'BZPJ-YGXG-DZ_D_CLS':  [
        'BZPJ-CLS-YGXG-DZ_D-latest'
        ],
    'BZPJ-YGXG-DZ_E_CLS':  [
        'BZPJ-CLS-YGXG-DZ_E-latest'
        ],

    'BZPJ-YBG-W':  [
        'BZPJ-OD-W_A-latest',
        'BZPJ-OD-W_B-latest',
        ],
   'BZPJ-OD-COAL':  [
        'BZPJ-OD-COAL-latest',
        'BZPJ-OD-COAL_BIG-latest'
        ],
    'BZPJ-OD-HZ':  [
        'BZPJ-OD-HZ-latest',
        'BZPJ-OD-HZ_DL-latest'
        ],    
    'BZPJ-OD-KLQ':  [
        'BZPJ-OD-KLQ-latest'
        ],     
    'BZPJ-CLS-ZZ':  [
        'BZPJ-CLS-ZZ-latest'
        ],
}

import os

def get_task_models(task_name):
    results = []

    if not _task_models.has_key(task_name):
        return None

    models = _task_models[task_name]
    for model in models:
        items = model.rpartition('-')
        name = items[0]
        version = items[2]

        if version == 'latest':
            if name not in _model_version:
                return None
            version = _model_version[items[0]]
        
        results.append(name + '-' + version)

    return results