Skip to content

Commit e74f39a

Browse files
fix: clean code and complete docstrings
1 parent dd23f15 commit e74f39a

18 files changed

Lines changed: 145 additions & 70 deletions

tvm/cli.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,6 @@ def list_versions(limit: int):
177177
click.echo(click.style(name, fg=color))
178178

179179

180-
181180
@click.command(name="list")
182181
@click.option('-l', '--limit', default=10, help='number of `latest versions` to list')
183182
def list_versions_backup(limit: int):
@@ -520,12 +519,12 @@ def install_plugin(options):
520519
def uninstall_plugin(options):
521520
"""Use the package installer pip in current tutor version."""
522521
repository = VersionManagerGitRepository()
523-
uninstaller =TutorPluginUninstaller(repository=repository)
522+
uninstaller = TutorPluginUninstaller(repository=repository)
524523
options = list(options)
525524
options.insert(0, "uninstall")
526525
options.append("-y")
527526
uninstaller(options)
528-
527+
529528

530529
@click.group(
531530
name="config",

tvm/version_manager/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
"""Version manager init."""
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
"""Application init."""
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,14 @@
1+
"""Tutor plugin installer application."""
12
from tvm.version_manager.domain.version_manager_repository import VersionManagerRepository
23

34

45
class TutorPluginInstaller:
6+
"""Tutor plugin installer for version manager."""
7+
58
def __init__(self, repository: VersionManagerRepository) -> None:
9+
"""init."""
610
self.repository = repository
711

812
def __call__(self, options) -> None:
13+
"""call."""
914
self.repository.install_plugin(options)
Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,14 @@
1-
from tvm.version_manager.domain.tutor_version import TutorVersion
1+
"""Tutor plugin uninstaller application."""
22
from tvm.version_manager.domain.version_manager_repository import VersionManagerRepository
33

44

55
class TutorPluginUninstaller:
6+
"""Tutor plugin uninstaller for version manager."""
7+
68
def __init__(self, repository: VersionManagerRepository) -> None:
9+
"""init."""
710
self.repository = repository
811

912
def __call__(self, options) -> None:
13+
"""call."""
1014
self.repository.uninstall_plugin(options)
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,16 @@
1+
"""Tutor version enabler application."""
12
from tvm.version_manager.domain.tutor_version import TutorVersion
23
from tvm.version_manager.domain.version_manager_repository import VersionManagerRepository
34

45

56
class TutorVersionEnabler:
7+
"""Tutor version enabler for version manager."""
8+
69
def __init__(self, repository: VersionManagerRepository) -> None:
10+
"""init."""
711
self.repository = repository
812

913
def __call__(self, version: str) -> None:
14+
"""call."""
1015
version = TutorVersion(version)
1116
self.repository.use_version(version)
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,16 @@
1+
"""Tutor version finder application."""
12
from tvm.version_manager.domain.tutor_version import TutorVersion
23
from tvm.version_manager.domain.version_manager_repository import VersionManagerRepository
34

45

56
class TutorVersionFinder:
7+
"""Tutor version finder for version manager."""
8+
69
def __init__(self, repository: VersionManagerRepository):
10+
"""init."""
711
self.repository = repository
812

913
def __call__(self, version: str) -> TutorVersion:
14+
"""call."""
1015
version = TutorVersion(version)
1116
return self.repository.find_version(version=version)
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,16 @@
1+
"""Tutor version installer application."""
12
from tvm.version_manager.domain.tutor_version import TutorVersion
23
from tvm.version_manager.domain.version_manager_repository import VersionManagerRepository
34

45

56
class TutorVersionInstaller:
7+
"""Tutor version installer for version manager."""
8+
69
def __init__(self, repository: VersionManagerRepository) -> None:
10+
"""init."""
711
self.repository = repository
812

913
def __call__(self, version: str) -> None:
14+
"""call."""
1015
version = TutorVersion(version)
1116
self.repository.install_version(version=version)
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,16 @@
1+
"""Tutor version uninstaller application."""
12
from tvm.version_manager.domain.tutor_version import TutorVersion
23
from tvm.version_manager.domain.version_manager_repository import VersionManagerRepository
34

45

56
class TutorVersionUninstaller:
7+
"""Tutor version uninstaller for version manager."""
8+
69
def __init__(self, repository: VersionManagerRepository) -> None:
10+
"""init."""
711
self.repository = repository
812

913
def __call__(self, version: str) -> None:
14+
"""call."""
1015
version = TutorVersion(version)
1116
self.repository.uninstall_version(version)
Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,17 @@
1+
"""Tutor version application."""
12
from typing import List
3+
24
from tvm.version_manager.domain.tutor_version import TutorVersion
3-
from tvm.version_manager.domain.version_manager_repository import (
4-
VersionManagerRepository,
5-
)
5+
from tvm.version_manager.domain.version_manager_repository import VersionManagerRepository
66

77

88
class TutorVersionLister:
9+
"""Tutor version lister for version manage."""
10+
911
def __init__(self, repository: VersionManagerRepository) -> None:
12+
"""init."""
1013
self.repository = repository
1114

1215
def __call__(self, limit: int) -> List[TutorVersion]:
16+
"""call."""
1317
return self.repository.list_versions(limit=limit)

0 commit comments

Comments
 (0)