SIAnalytics · nijkah · Sep 8, 2022 · Sep 8, 2022
diff --git a/.github/workflows/build_docker.yaml b/.github/workflows/build_docker.yaml
@@ -2,7 +2,7 @@ name: Build & Push Docker Image
 on:
   push:
     branches:
-      - 'master'
+      - 'main'
       - 'test/*'
     tags:
       - '**'
@@ -14,7 +14,7 @@ concurrency:
 env:
   DOCKERFILE_PATH: "./docker/Dockerfile"
   PROJECT_NAME: "mlops"
-  APP_NAME: "mmtune"
+  APP_NAME: "siatune"
 
 jobs:
   build:

diff --git a/.github/workflows/build_unit_test.yaml b/.github/workflows/build_unit_test.yaml
@@ -3,7 +3,7 @@ name: Build & Unit Test
 on:
   pull_request:
     branch:
-        - 'master'
+        - 'main'
     paths-ignore:
       - ".github/**.md"
       - "README.md"
@@ -31,7 +31,7 @@ jobs:
     - name: Run unittests and generate coverage report
       run: |
           pip install pytest coverage
-          coverage run --branch --source mmtune -m pytest tests/
+          coverage run --branch --source siatune -m pytest tests/
           coverage xml
           coverage report -m
     - name: Display coverage

diff --git a/.github/workflows/publish-to-pypi.yaml b/.github/workflows/publish-to-pypi.yaml
@@ -24,6 +24,7 @@ jobs:
       env:
           PYPI_USER: ${{ secrets.PYPI_USER }}
           PYPI_PW: ${{ secrets.PYPI_PW}}
+          PIP_INDEX_URL: ${{ secrets.PIP_INDEX_URL}}
       run: |
         pip3 install twine
-        twine upload --repository-url https://pypi.sia-service.kr/ dist/* -u ${PYPI_USER} -p ${PYPI_PW}
+        twine upload --repository-url ${PIP_INDEX_URL} dist/* -u ${PYPI_USER} -p ${PYPI_PW}
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -41,4 +41,4 @@ repos:
     rev: v0.1.0  # Use the ref you want to point at
     hooks:
       - id: check-copyright
-        args: ["mmtune"]  # replace the dir_to_check with your expected directory to check
+        args: ["siatune"]  # replace the dir_to_check with your expected directory to check
diff --git a/README.md b/README.md
@@ -1,12 +1,12 @@
 
 <div align="center">
-  <img src="resources/mmtune-logo.png" width="450"/>
+  <img src="resources/siatune-logo.png" width="450"/>
 </div>
 
 ## Introduction
-MMTune is an open-source deep learning model hyperparameter tuning toolbox especially for OpenMMLab's model frameworks such as [mmdetection](https://github.com/open-mmlab/mmdetection) and [mmsegmentation](https://github.com/open-mmlab/mmsegmentation). In order to support job scheduling and resource management, MMTune adopts [Ray](https://github.com/ray-project/ray) and [Ray.tune](https://docs.ray.io/en/latest/tune/index.html).
+SIATune is an open-source deep learning model hyperparameter tuning toolbox especially for OpenMMLab's model frameworks such as [mmdetection](https://github.com/open-mmlab/mmdetection) and [mmsegmentation](https://github.com/open-mmlab/mmsegmentation). In order to support job scheduling and resource management, SIATune adopts [Ray](https://github.com/ray-project/ray) and [Ray.tune](https://docs.ray.io/en/latest/tune/index.html).
 
-Currently, MMTune is maintained by SI-Analytics.
+Currently, SIATune is maintained by SI-Analytics.
 
 ### Major features
 

diff --git a/configs/mmtune/_base_/callback/mlflow.py b/configs/mmtune/_base_/callback/mlflow.py
@@ -1,7 +1,7 @@
 callbacks = [
     dict(
         type='MLflowLoggerCallback',
-        experiment_name='mmtune',
+        experiment_name='siatune',
         save_artifact=True,
         metric='train/loss',
         mode='max',

diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -23,7 +23,7 @@ ARG CUDA
 ARG MMCV_VERSION
 RUN ["/bin/bash", "-c", "pip install --no-cache-dir mmcv-full==${MMCV_VERSION} -f https://download.openmmlab.com/mmcv/dist/cu${CUDA//./}/torch${PYTORCH}/index.html"]
 
-# Install MMTune
-COPY . /mmtune
-WORKDIR /mmtune
+# Install SIATune
+COPY . /siatune
+WORKDIR /siatune
 RUN pip install --no-cache-dir -e .
diff --git a/docs/get_started.md b/docs/get_started.md
@@ -1,8 +1,8 @@
-# Build MMTune
+# Build SIATune
 
 ## Dockerfile (RECOMMENDED)
 ```bash
-docker build . -t mmtune:master -f docker/Dockerfile
+docker build . -t siatune:master -f docker/Dockerfile
 ```
 
 ## Build From Source
@@ -12,10 +12,10 @@ docker build . -t mmtune:master -f docker/Dockerfile
 pip install torch torchvision -f https://download.pytorch.org/whl/torch_stable.html
 # 2. install mmcv
 pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html
-# 3. clone mmtune
-git clone -b master https://github.com/SIAnalytics/mmtune.git mmtune
-# 4. install mmtune
-cd mmtune && pip install -e .
+# 3. clone siatune
+git clone -b master https://github.com/SIAnalytics/siatune.git siatune
+# 4. install siatune
+cd siatune && pip install -e .
 ```
 
 # Hyperparamer tuning with OpenMMLab's model frameworks.
@@ -34,5 +34,5 @@ python tools/tune.py ${TUNE_CONFIG} [optional tune arguments] [optional task arg
 
 ```bash
 # MMDetection Example
-python tools/tune.py configs/mmtune/mmdet_asynchb_nevergrad_pso.py --trainable_args configs/mmdet/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py
+python tools/tune.py configs/siatune/mmdet_asynchb_nevergrad_pso.py --trainable_args configs/mmdet/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py
 ```
diff --git a/resources/mmtune-logo.png b/resources/mmtune-logo.png
diff --git a/resources/siatune-logo.png b/resources/siatune-logo.png
diff --git a/setup.cfg b/setup.cfg
@@ -12,7 +12,7 @@ SPLIT_BEFORE_EXPRESSION_AFTER_OPENING_PAREN = true
 line_length = 79
 multi_line_output = 0
 extra_standard_library = setuptools
-known_first_party = mmtune
+known_first_party = siatune
 known_third_party = mmcv,numpy,pytest,ray,setuptools,torch
 no_lines_before = STDLIB,LOCALFOLDER
 default_section = THIRDPARTY
diff --git a/setup.py b/setup.py
@@ -1,6 +1,6 @@
 from setuptools import find_packages, setup
 
-version_file = 'mmtune/version.py'
+version_file = 'siatune/version.py'
 
 
 def parse_requirements(fname: str = 'requirements.txt',
@@ -113,7 +113,7 @@ def get_version() -> str:
 
 if __name__ == '__main__':
     setup(
-        name='mmtune',
+        name='siatune',
         version=get_version(),
         author='SIA',
         packages=find_packages(exclude=('configs', 'tools')),

diff --git a/mmtune/__init__.py → siatune/__init__.py b/mmtune/__init__.py → siatune/__init__.py
diff --git a/mmtune/__main__.py → siatune/__main__.py b/mmtune/__main__.py → siatune/__main__.py
@@ -1,5 +1,5 @@
 # Copyright (c) SI-Analytics. All rights reserved.
-"""The mmtune entry point."""
+"""The siatune entry point."""
 from .run import main
 
 if __name__ == '__main__':

diff --git a/mmtune/apis/__init__.py → siatune/apis/__init__.py b/mmtune/apis/__init__.py → siatune/apis/__init__.py
diff --git a/mmtune/apis/analysis.py → siatune/apis/analysis.py b/mmtune/apis/analysis.py → siatune/apis/analysis.py
@@ -7,7 +7,7 @@
 from mmcv.utils import Config, get_logger
 from ray import tune
 
-from mmtune.utils import ImmutableContainer, dump_cfg
+from siatune.utils import ImmutableContainer, dump_cfg
 
 
 def log_analysis(analysis: tune.ExperimentAnalysis,
@@ -31,7 +31,7 @@ def log_analysis(analysis: tune.ExperimentAnalysis,
 
     timestamp = time.strftime('%Y%m%d_%H%M%S', time.localtime())
     logger = get_logger(
-        'mmtune', log_file=osp.join(log_dir, f'{timestamp}.log'))
+        'siatune', log_file=osp.join(log_dir, f'{timestamp}.log'))
 
     logger.info(
         f'Best Hyperparam: \n'

diff --git a/mmtune/apis/tune.py → siatune/apis/tune.py b/mmtune/apis/tune.py → siatune/apis/tune.py
@@ -5,12 +5,12 @@
 import ray
 from mmcv.utils import Config
 
-from mmtune.mm.tasks import BaseTask
-from mmtune.ray.callbacks import build_callback
-from mmtune.ray.schedulers import build_scheduler
-from mmtune.ray.searchers import build_searcher
-from mmtune.ray.spaces import build_space
-from mmtune.ray.stoppers import build_stopper
+from siatune.mm.tasks import BaseTask
+from siatune.ray.callbacks import build_callback
+from siatune.ray.schedulers import build_scheduler
+from siatune.ray.searchers import build_searcher
+from siatune.ray.spaces import build_space
+from siatune.ray.stoppers import build_stopper
 
 
 def tune(task_processor: BaseTask, tune_config: Config,

diff --git a/mmtune/mm/__init__.py → siatune/mm/__init__.py b/mmtune/mm/__init__.py → siatune/mm/__init__.py
diff --git a/mmtune/mm/context/__init__.py → siatune/mm/context/__init__.py b/mmtune/mm/context/__init__.py → siatune/mm/context/__init__.py
diff --git a/mmtune/mm/context/manager.py → siatune/mm/context/manager.py b/mmtune/mm/context/manager.py → siatune/mm/context/manager.py
diff --git a/mmtune/mm/context/rewriters/__init__.py → siatune/mm/context/rewriters/__init__.py b/mmtune/mm/context/rewriters/__init__.py → siatune/mm/context/rewriters/__init__.py
diff --git a/mmtune/mm/context/rewriters/base.py → siatune/mm/context/rewriters/base.py b/mmtune/mm/context/rewriters/base.py → siatune/mm/context/rewriters/base.py
diff --git a/mmtune/mm/context/rewriters/builder.py → siatune/mm/context/rewriters/builder.py b/mmtune/mm/context/rewriters/builder.py → siatune/mm/context/rewriters/builder.py
diff --git a/mmtune/mm/context/rewriters/dump.py → siatune/mm/context/rewriters/dump.py b/mmtune/mm/context/rewriters/dump.py → siatune/mm/context/rewriters/dump.py
@@ -5,7 +5,7 @@
 
 import ray
 
-from mmtune.utils import dump_cfg
+from siatune.utils import dump_cfg
 from .base import BaseRewriter
 from .builder import REWRITERS
 

diff --git a/mmtune/mm/context/rewriters/instantiate.py → siatune/mm/context/rewriters/instantiate.py b/mmtune/mm/context/rewriters/instantiate.py → siatune/mm/context/rewriters/instantiate.py
diff --git a/mmtune/mm/context/rewriters/merge.py → siatune/mm/context/rewriters/merge.py b/mmtune/mm/context/rewriters/merge.py → siatune/mm/context/rewriters/merge.py
diff --git a/mmtune/mm/context/rewriters/patch.py → siatune/mm/context/rewriters/patch.py b/mmtune/mm/context/rewriters/patch.py → siatune/mm/context/rewriters/patch.py
diff --git a/mmtune/mm/context/rewriters/path.py → siatune/mm/context/rewriters/path.py b/mmtune/mm/context/rewriters/path.py → siatune/mm/context/rewriters/path.py
diff --git a/mmtune/mm/context/rewriters/register.py → siatune/mm/context/rewriters/register.py b/mmtune/mm/context/rewriters/register.py → siatune/mm/context/rewriters/register.py
@@ -21,7 +21,7 @@ def __init__(self,
             hk_key (str): The key of the custom hooks.
         """
         # Re-register the hook in the new process.
-        from mmtune.mm import hooks  # noqa F401
+        from siatune.mm import hooks  # noqa F401
 
         self.post_custom_hooks = post_custom_hooks
         self.key = key

diff --git a/mmtune/mm/context/rewriters/resume.py → siatune/mm/context/rewriters/resume.py b/mmtune/mm/context/rewriters/resume.py → siatune/mm/context/rewriters/resume.py
diff --git a/mmtune/mm/hooks/__init__.py → siatune/mm/hooks/__init__.py b/mmtune/mm/hooks/__init__.py → siatune/mm/hooks/__init__.py
diff --git a/mmtune/mm/hooks/checkpoint.py → siatune/mm/hooks/checkpoint.py b/mmtune/mm/hooks/checkpoint.py → siatune/mm/hooks/checkpoint.py
diff --git a/mmtune/mm/hooks/reporter.py → siatune/mm/hooks/reporter.py b/mmtune/mm/hooks/reporter.py → siatune/mm/hooks/reporter.py
diff --git a/mmtune/mm/tasks/__init__.py → siatune/mm/tasks/__init__.py b/mmtune/mm/tasks/__init__.py → siatune/mm/tasks/__init__.py
diff --git a/mmtune/mm/tasks/base.py → siatune/mm/tasks/base.py b/mmtune/mm/tasks/base.py → siatune/mm/tasks/base.py
@@ -6,8 +6,8 @@
 
 import ray
 
-from mmtune.mm.context import ContextManager
-from mmtune.utils import ImmutableContainer
+from siatune.mm.context import ContextManager
+from siatune.utils import ImmutableContainer
 
 
 class BaseTask(metaclass=ABCMeta):

diff --git a/mmtune/mm/tasks/blackbox.py → siatune/mm/tasks/blackbox.py b/mmtune/mm/tasks/blackbox.py → siatune/mm/tasks/blackbox.py
diff --git a/mmtune/mm/tasks/builder.py → siatune/mm/tasks/builder.py b/mmtune/mm/tasks/builder.py → siatune/mm/tasks/builder.py
diff --git a/mmtune/mm/tasks/cont_test_func.py → siatune/mm/tasks/cont_test_func.py b/mmtune/mm/tasks/cont_test_func.py → siatune/mm/tasks/cont_test_func.py
diff --git a/mmtune/mm/tasks/disc_test_func.py → siatune/mm/tasks/disc_test_func.py b/mmtune/mm/tasks/disc_test_func.py → siatune/mm/tasks/disc_test_func.py
diff --git a/mmtune/mm/tasks/mmcls.py → siatune/mm/tasks/mmcls.py b/mmtune/mm/tasks/mmcls.py → siatune/mm/tasks/mmcls.py
diff --git a/mmtune/mm/tasks/mmdet.py → siatune/mm/tasks/mmdet.py b/mmtune/mm/tasks/mmdet.py → siatune/mm/tasks/mmdet.py
diff --git a/mmtune/mm/tasks/mmseg.py → siatune/mm/tasks/mmseg.py b/mmtune/mm/tasks/mmseg.py → siatune/mm/tasks/mmseg.py
diff --git a/mmtune/mm/tasks/mmtrainbase.py → siatune/mm/tasks/mmtrainbase.py b/mmtune/mm/tasks/mmtrainbase.py → siatune/mm/tasks/mmtrainbase.py
diff --git a/mmtune/ray/__init__.py → siatune/ray/__init__.py b/mmtune/ray/__init__.py → siatune/ray/__init__.py
diff --git a/mmtune/ray/callbacks/__init__.py → siatune/ray/callbacks/__init__.py b/mmtune/ray/callbacks/__init__.py → siatune/ray/callbacks/__init__.py
diff --git a/mmtune/ray/callbacks/builder.py → siatune/ray/callbacks/builder.py b/mmtune/ray/callbacks/builder.py → siatune/ray/callbacks/builder.py
diff --git a/mmtune/ray/callbacks/mlflow.py → siatune/ray/callbacks/mlflow.py b/mmtune/ray/callbacks/mlflow.py → siatune/ray/callbacks/mlflow.py
diff --git a/mmtune/ray/schedulers/__init__.py → siatune/ray/schedulers/__init__.py b/mmtune/ray/schedulers/__init__.py → siatune/ray/schedulers/__init__.py
diff --git a/mmtune/ray/schedulers/builder.py → siatune/ray/schedulers/builder.py b/mmtune/ray/schedulers/builder.py → siatune/ray/schedulers/builder.py
diff --git a/mmtune/ray/schedulers/pbt.py → siatune/ray/schedulers/pbt.py b/mmtune/ray/schedulers/pbt.py → siatune/ray/schedulers/pbt.py
@@ -8,9 +8,9 @@
     PopulationBasedTraining as _PopulationBasedTraining
 from ray.tune.trial import Trial
 
-from mmtune.ray.schedulers import SCHEDULERS
-from mmtune.ray.spaces import build_space
-from mmtune.utils import ImmutableContainer
+from siatune.ray.schedulers import SCHEDULERS
+from siatune.ray.spaces import build_space
+from siatune.utils import ImmutableContainer
 
 
 def explore(

diff --git a/mmtune/ray/searchers/__init__.py → siatune/ray/searchers/__init__.py b/mmtune/ray/searchers/__init__.py → siatune/ray/searchers/__init__.py
diff --git a/mmtune/ray/searchers/builder.py → siatune/ray/searchers/builder.py b/mmtune/ray/searchers/builder.py → siatune/ray/searchers/builder.py
diff --git a/mmtune/ray/searchers/flaml.py → siatune/ray/searchers/flaml.py b/mmtune/ray/searchers/flaml.py → siatune/ray/searchers/flaml.py
diff --git a/mmtune/ray/searchers/hyperopt.py → siatune/ray/searchers/hyperopt.py b/mmtune/ray/searchers/hyperopt.py → siatune/ray/searchers/hyperopt.py
diff --git a/mmtune/ray/searchers/nevergrad.py → siatune/ray/searchers/nevergrad.py b/mmtune/ray/searchers/nevergrad.py → siatune/ray/searchers/nevergrad.py
diff --git a/mmtune/ray/spaces/__init__.py → siatune/ray/spaces/__init__.py b/mmtune/ray/spaces/__init__.py → siatune/ray/spaces/__init__.py
diff --git a/mmtune/ray/spaces/base.py → siatune/ray/spaces/base.py b/mmtune/ray/spaces/base.py → siatune/ray/spaces/base.py
diff --git a/mmtune/ray/spaces/builder.py → siatune/ray/spaces/builder.py b/mmtune/ray/spaces/builder.py → siatune/ray/spaces/builder.py
diff --git a/mmtune/ray/spaces/choice.py → siatune/ray/spaces/choice.py b/mmtune/ray/spaces/choice.py → siatune/ray/spaces/choice.py
@@ -3,7 +3,7 @@
 
 import ray.tune as tune
 
-from mmtune.utils import ImmutableContainer
+from siatune.utils import ImmutableContainer
 from .base import BaseSpace
 from .builder import SPACES
 

diff --git a/mmtune/ray/spaces/grid_search.py → siatune/ray/spaces/grid_search.py b/mmtune/ray/spaces/grid_search.py → siatune/ray/spaces/grid_search.py
@@ -3,7 +3,7 @@
 
 import ray.tune as tune
 
-from mmtune.utils import ImmutableContainer
+from siatune.utils import ImmutableContainer
 from .base import BaseSpace
 from .builder import SPACES
 

diff --git a/mmtune/ray/spaces/sample_from.py → siatune/ray/spaces/sample_from.py b/mmtune/ray/spaces/sample_from.py → siatune/ray/spaces/sample_from.py
diff --git a/mmtune/ray/stoppers/__init__.py → siatune/ray/stoppers/__init__.py b/mmtune/ray/stoppers/__init__.py → siatune/ray/stoppers/__init__.py
diff --git a/mmtune/ray/stoppers/builder.py → siatune/ray/stoppers/builder.py b/mmtune/ray/stoppers/builder.py → siatune/ray/stoppers/builder.py
diff --git a/mmtune/ray/stoppers/dict_stop.py → siatune/ray/stoppers/dict_stop.py b/mmtune/ray/stoppers/dict_stop.py → siatune/ray/stoppers/dict_stop.py
diff --git a/mmtune/ray/stoppers/early_drop.py → siatune/ray/stoppers/early_drop.py b/mmtune/ray/stoppers/early_drop.py → siatune/ray/stoppers/early_drop.py
diff --git a/mmtune/run.py → siatune/run.py b/mmtune/run.py → siatune/run.py
@@ -6,8 +6,8 @@
 import ray
 from mmcv import Config
 
-from mmtune.apis import log_analysis, tune
-from mmtune.mm.tasks import build_task_processor
+from siatune.apis import log_analysis, tune
+from siatune.mm.tasks import build_task_processor
 
 
 def parse_args() -> Namespace:

diff --git a/mmtune/utils/__init__.py → siatune/utils/__init__.py b/mmtune/utils/__init__.py → siatune/utils/__init__.py
diff --git a/mmtune/utils/config.py → siatune/utils/config.py b/mmtune/utils/config.py → siatune/utils/config.py
@@ -18,7 +18,7 @@ def dump_cfg(cfg: Config, save_path: str) -> bool:
             f.write(cfg.pretty_text)
         return True
     except Exception as err:
-        from mmtune.utils import get_root_logger
+        from siatune.utils import get_root_logger
         logger = get_root_logger()
         logger.error(f'Failed to dump config to {save_path}: {err}')
         return False
diff --git a/mmtune/utils/container.py → siatune/utils/container.py b/mmtune/utils/container.py → siatune/utils/container.py
diff --git a/mmtune/utils/logger.py → siatune/utils/logger.py b/mmtune/utils/logger.py → siatune/utils/logger.py
@@ -10,7 +10,7 @@ def get_root_logger(log_file: Optional[str] = None,
     """Get the root logger. The logger will be initialized if it has not been
     initialized. By default a StreamHandler will be added. If `log_file` is
     specified, a FileHandler will also be added. The name of the root logger is
-    the top-level package name, e.g., "mmtune".
+    the top-level package name, e.g., "siatune".
 
     Args:
         log_file (Optional[str]): The log filename. If specified, a FileHandler
@@ -22,6 +22,6 @@ def get_root_logger(log_file: Optional[str] = None,
         logging.Logger: The root logger.
     """
 
-    logger = get_logger(name='mmtune', log_file=log_file, log_level=log_level)
+    logger = get_logger(name='siatune', log_file=log_file, log_level=log_level)
 
     return logger
diff --git a/mmtune/version.py → siatune/version.py b/mmtune/version.py → siatune/version.py
diff --git a/tests/test_apis/test_apis.py b/tests/test_apis/test_apis.py
@@ -5,7 +5,7 @@
 import mmcv
 from ray.tune.trainable import Trainable
 
-from mmtune.apis import log_analysis, tune
+from siatune.apis import log_analysis, tune
 
 
 def test_log_analysis():

diff --git a/tests/test_mm/test_context.py b/tests/test_mm/test_context.py
@@ -1,6 +1,6 @@
 import pytest
 
-from mmtune.mm.context import REWRITERS, ContextManager
+from siatune.mm.context import REWRITERS, ContextManager
 
 
 def test_contextmanager():

diff --git a/tests/test_mm/test_hooks.py b/tests/test_mm/test_hooks.py
@@ -6,7 +6,7 @@
 import torch.distributed as dist
 from ray import tune
 
-from mmtune.mm.hooks import RayCheckpointHook, RayTuneLoggerHook
+from siatune.mm.hooks import RayCheckpointHook, RayTuneLoggerHook
 
 
 def test_raycheckpointhook():

diff --git a/tests/test_mm/test_rewriters.py b/tests/test_mm/test_rewriters.py
@@ -5,14 +5,14 @@
 import mmcv
 import pytest
 
-from mmtune.mm.context.rewriters import (REWRITERS, AppendTrialIDtoPath,
-                                         BaseRewriter, BatchConfigPatcher,
-                                         CustomHookRegister, Dump,
-                                         InstantiateCfg, MergeConfig,
-                                         ResumeFromCkpt,
-                                         SequeunceConfigPatcher)
-from mmtune.mm.context.rewriters.builder import build_rewriter
-from mmtune.utils import dump_cfg
+from siatune.mm.context.rewriters import (REWRITERS, AppendTrialIDtoPath,
+                                          BaseRewriter, BatchConfigPatcher,
+                                          CustomHookRegister, Dump,
+                                          InstantiateCfg, MergeConfig,
+                                          ResumeFromCkpt,
+                                          SequeunceConfigPatcher)
+from siatune.mm.context.rewriters.builder import build_rewriter
+from siatune.utils import dump_cfg
 
 
 def test_base_rewriter():
@@ -78,7 +78,7 @@ def test_merge():
 
 
 def test_patch():
-    from mmtune.mm.context.rewriters.patch import unwrap_regexp
+    from siatune.mm.context.rewriters.patch import unwrap_regexp
     assert unwrap_regexp('$(a)') == ('a', True)
 
     context = dict(

diff --git a/tests/test_mm/test_tasks.py b/tests/test_mm/test_tasks.py
@@ -7,11 +7,11 @@
 from mmcv.utils import Config
 from ray import tune
 
-from mmtune.mm.tasks import (TASKS, BaseTask, BlackBoxTask,
-                             ContinuousTestFunction, DiscreteTestFunction,
-                             MMClassification, MMDetection, MMSegmentation,
-                             MMTrainBasedTask, build_task_processor)
-from mmtune.utils.config import dump_cfg
+from siatune.mm.tasks import (TASKS, BaseTask, BlackBoxTask,
+                              ContinuousTestFunction, DiscreteTestFunction,
+                              MMClassification, MMDetection, MMSegmentation,
+                              MMTrainBasedTask, build_task_processor)
+from siatune.utils.config import dump_cfg
 
 _session = dict()
 

diff --git a/tests/test_ray/test_schedulers.py b/tests/test_ray/test_schedulers.py
@@ -1,4 +1,4 @@
-from mmtune.ray.schedulers import SCHEDULERS, build_scheduler
+from siatune.ray.schedulers import SCHEDULERS, build_scheduler
 
 
 def test_build_schedulers():

diff --git a/tests/test_ray/test_searchers.py b/tests/test_ray/test_searchers.py
@@ -1,9 +1,9 @@
 import pytest
 from ray import tune
 
-from mmtune.ray.searchers import (SEARCHERS, BlendSearch, CFOSearch,
-                                  HyperOptSearch, NevergradSearch,
-                                  build_searcher)
+from siatune.ray.searchers import (SEARCHERS, BlendSearch, CFOSearch,
+                                   HyperOptSearch, NevergradSearch,
+                                   build_searcher)
 
 
 def test_build_searcher():

diff --git a/tests/test_ray/test_spaces.py b/tests/test_ray/test_spaces.py
@@ -1,8 +1,8 @@
 import pytest
 from ray import tune
 
-from mmtune.ray.spaces import Choice, GridSearch, SampleFrom, build_space
-from mmtune.utils import ImmutableContainer
+from siatune.ray.spaces import Choice, GridSearch, SampleFrom, build_space
+from siatune.utils import ImmutableContainer
 
 
 def test_build_space():