# Path: astroNN/lamost/lamost_shared.py
# def lamost_default_dr(dr=None):
#     """
#     Check if dr argument is provided, if none then use default
# 
#     :param dr: data release
#     :type dr: Union(int, NoneType)
#     :return: data release
#     :rtype: int
#     :History: 2018-May-13 - Written - Henry Leung (University of Toronto)
#     """
#     # enforce dr5 restriction
#     if dr is None:
#         dr = 5
#         print(f'dr is not provided, using default dr={dr}')
#     elif dr == 5:
#         pass
#     else:
#         raise ValueError('Only LAMOST DR5 is supported')
# 
#     return dr


import numpy as np
from astroNN.lamost.lamost_shared import lamost_default_dr



def wavelength_solution(dr=None):
    """
    To return wavelegnth_solution

    :param dr: data release
    :type dr: Union(int, NoneType)
    :return: wavelength solution array
    :rtype: ndarray
    :History: 2018-Mar-15 - Written - Henry Leung (University of Toronto)
    """

# Path: astroNN/lamost/lamost_shared.py
# def lamost_default_dr(dr=None):
#     """
#     Check if dr argument is provided, if none then use default
# 
#     :param dr: data release
#     :type dr: Union(int, NoneType)
#     :return: data release
#     :rtype: int
#     :History: 2018-May-13 - Written - Henry Leung (University of Toronto)
#     """
#     # enforce dr5 restriction
#     if dr is None:
#         dr = 5
#         print(f'dr is not provided, using default dr={dr}')
#     elif dr == 5:
#         pass
#     else:
#         raise ValueError('Only LAMOST DR5 is supported')
# 
#     return dr


# Path: astroNN/lamost/chips.py
import numpy as np
from astroNN.lamost.lamost_shared import lamost_default_dr




def wavelength_solution(dr=None):
    """
    To return wavelegnth_solution

    :param dr: data release
    :type dr: Union(int, NoneType)
    :return: wavelength solution array
    :rtype: ndarray
    :History: 2018-Mar-15 - Written - Henry Leung (University of Toronto)
    """

    lamost_default_dr(dr=dr)