# Copyright 2021 cstsunfu. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""schedulers"""
import importlib
import os
from dlk.utils.register import Register
from torch.optim import Optimizer
from torch.optim.lr_scheduler import LambdaLR
import math


scheduler_config_register = Register("Schedule config register.")
scheduler_register = Register("Schedule register.")


class BaseScheduler(object):
    """interface for Schedule"""

    def get_scheduler(self)->LambdaLR:
        """return the initialized scheduler

        Returns: 
            Schedule

        """
        raise NotImplementedError

    def __call__(self):
        """the same as self.get_scheduler()
        """
        return self.get_scheduler()


def import_schedulers(schedulers_dir, namespace):
    for file in os.listdir(schedulers_dir):
        path = os.path.join(schedulers_dir, file)
        if (
            not file.startswith("_")
            and not file.startswith(".")
            and (file.endswith(".py") or os.path.isdir(path))
        ):
            scheduler_name = file[: file.find(".py")] if file.endswith(".py") else file
            importlib.import_module(namespace + "." + scheduler_name)


# automatically import any Python files in the schedulers directory
schedulers_dir = os.path.dirname(__file__)
import_schedulers(schedulers_dir, "dlk.core.schedulers")

[((801, 838), 'dlk.utils.register.Register', 'Register', (['"""Schedule config register."""'], {}), "('Schedule config register.')\n", (809, 838), False, 'from dlk.utils.register import Register\n'), ((860, 890), 'dlk.utils.register.Register', 'Register', (['"""Schedule register."""'], {}), "('Schedule register.')\n", (868, 890), False, 'from dlk.utils.register import Register\n'), ((1798, 1823), 'os.path.dirname', 'os.path.dirname', (['__file__'], {}), '(__file__)\n', (1813, 1823), False, 'import os\n'), ((1311, 1337), 'os.listdir', 'os.listdir', (['schedulers_dir'], {}), '(schedulers_dir)\n', (1321, 1337), False, 'import os\n'), ((1354, 1388), 'os.path.join', 'os.path.join', (['schedulers_dir', 'file'], {}), '(schedulers_dir, file)\n', (1366, 1388), False, 'import os\n'), ((1653, 1710), 'importlib.import_module', 'importlib.import_module', (["(namespace + '.' + scheduler_name)"], {}), "(namespace + '.' + scheduler_name)\n", (1676, 1710), False, 'import importlib\n'), ((1521, 1540), 'os.path.isdir', 'os.path.isdir', (['path'], {}), '(path)\n', (1534, 1540), False, 'import os\n')]