dstack 0.19.25__py3-none-any.whl → 0.19.26__py3-none-any.whl

dstack/_internal/cli/commands/__init__.py

@@ -54,7 +54,7 @@ class BaseCommand(ABC):
 
 
 class APIBaseCommand(BaseCommand):
-    api: Client = None
+    api: Client
 
     def _register(self):
         self._parser.add_argument(
@@ -62,7 +62,7 @@ class APIBaseCommand(BaseCommand):
             help="The name of the project. Defaults to [code]$DSTACK_PROJECT[/]",
             metavar="NAME",
             default=os.getenv("DSTACK_PROJECT"),
-        ).completer = ProjectNameCompleter()
+        ).completer = ProjectNameCompleter()  # type: ignore[attr-defined]
 
     def _command(self, args: argparse.Namespace):
         configure_logging()

dstack/_internal/cli/commands/apply.py

@@ -1,7 +1,6 @@
 import argparse
-from pathlib import Path
 
-from argcomplete import FilesCompleter
+from argcomplete import FilesCompleter  # type: ignore[attr-defined]
 
 from dstack._internal.cli.commands import APIBaseCommand
 from dstack._internal.cli.services.configurators import (
@@ -9,12 +8,7 @@ from dstack._internal.cli.services.configurators import (
     get_apply_configurator_class,
     load_apply_configuration,
 )
-from dstack._internal.cli.services.repos import (
-    init_default_virtual_repo,
-    init_repo,
-    register_init_repo_args,
-)
-from dstack._internal.cli.utils.common import console, warn
+from dstack._internal.cli.utils.common import console
 from dstack._internal.core.errors import CLIError
 from dstack._internal.core.models.configurations import ApplyConfigurationType
 
@@ -48,7 +42,7 @@ class ApplyCommand(APIBaseCommand):
                 " Defaults to [code]$PWD/.dstack.yml[/]"
             ),
             dest="configuration_file",
-        ).completer = FilesCompleter(allowednames=["*.yml", "*.yaml"])
+        ).completer = FilesCompleter(allowednames=["*.yml", "*.yaml"])  # type: ignore[attr-defined]
         self._parser.add_argument(
             "-y",
             "--yes",
@@ -66,37 +60,6 @@ class ApplyCommand(APIBaseCommand):
             help="Exit immediately after submitting configuration",
             action="store_true",
         )
-        self._parser.add_argument(
-            "--ssh-identity",
-            metavar="SSH_PRIVATE_KEY",
-            help="The private SSH key path for SSH tunneling",
-            type=Path,
-            dest="ssh_identity_file",
-        )
-        repo_group = self._parser.add_argument_group("Repo Options")
-        repo_group.add_argument(
-            "-P",
-            "--repo",
-            help=("The repo to use for the run. Can be a local path or a Git repo URL."),
-            dest="repo",
-        )
-        repo_group.add_argument(
-            "--repo-branch",
-            help="The repo branch to use for the run",
-            dest="repo_branch",
-        )
-        repo_group.add_argument(
-            "--repo-hash",
-            help="The hash of the repo commit to use for the run",
-            dest="repo_hash",
-        )
-        repo_group.add_argument(
-            "--no-repo",
-            help="Do not use any repo for the run",
-            dest="no_repo",
-            action="store_true",
-        )
-        register_init_repo_args(repo_group)
 
     def _command(self, args: argparse.Namespace):
         try:
@@ -117,26 +80,6 @@ class ApplyCommand(APIBaseCommand):
             super()._command(args)
             if not args.yes and args.configuration_file == APPLY_STDIN_NAME:
                 raise CLIError("Cannot read configuration from stdin if -y/--yes is not specified")
-            if args.repo and args.no_repo:
-                raise CLIError("Either --repo or --no-repo can be specified")
-            if args.local:
-                warn(
-                    "Local repos are deprecated since 0.19.25 and will be removed soon."
-                    " Consider using `files` instead: https://dstack.ai/docs/concepts/tasks/#files"
-                )
-            repo = None
-            if args.repo:
-                repo = init_repo(
-                    api=self.api,
-                    repo_path=args.repo,
-                    repo_branch=args.repo_branch,
-                    repo_hash=args.repo_hash,
-                    local=args.local,
-                    git_identity_file=args.git_identity_file,
-                    oauth_token=args.gh_token,
-                )
-            elif args.no_repo:
-                repo = init_default_virtual_repo(api=self.api)
             configuration_path, configuration = load_apply_configuration(args.configuration_file)
             configurator_class = get_apply_configurator_class(configuration.type)
             configurator = configurator_class(api_client=self.api)
@@ -148,7 +91,6 @@ class ApplyCommand(APIBaseCommand):
                 command_args=args,
                 configurator_args=known,
                 unknown_args=unknown,
-                repo=repo,
             )
         except KeyboardInterrupt:
             console.print("\nOperation interrupted by user. Exiting...")

dstack/_internal/cli/commands/attach.py

@@ -61,7 +61,7 @@ class AttachCommand(APIBaseCommand):
             type=int,
             default=0,
         )
-        self._parser.add_argument("run_name").completer = RunNameCompleter()
+        self._parser.add_argument("run_name").completer = RunNameCompleter()  # type: ignore[attr-defined]
 
     def _command(self, args: argparse.Namespace):
         super()._command(args)

dstack/_internal/cli/commands/completion.py

@@ -17,4 +17,4 @@ class CompletionCommand(BaseCommand):
 
     def _command(self, args):
         super()._command(args)
-        print(argcomplete.shellcode(["dstack"], shell=args.shell))
+        print(argcomplete.shellcode(["dstack"], shell=args.shell))  # type: ignore[attr-defined]

dstack/_internal/cli/commands/delete.py

@@ -1,7 +1,7 @@
 import argparse
 from pathlib import Path
 
-from argcomplete import FilesCompleter
+from argcomplete import FilesCompleter  # type: ignore[attr-defined]
 
 from dstack._internal.cli.commands import APIBaseCommand
 from dstack._internal.cli.services.configurators import (
@@ -24,7 +24,7 @@ class DeleteCommand(APIBaseCommand):
             metavar="FILE",
             help="The path to the configuration file. Defaults to [code]$PWD/.dstack.yml[/]",
             dest="configuration_file",
-        ).completer = FilesCompleter(allowednames=["*.yml", "*.yaml"])
+        ).completer = FilesCompleter(allowednames=["*.yml", "*.yaml"])  # type: ignore[attr-defined]
         self._parser.add_argument(
             "-y",
             "--yes",

dstack/_internal/cli/commands/fleet.py

@@ -48,7 +48,7 @@ class FleetCommand(APIBaseCommand):
         delete_parser.add_argument(
             "name",
             help="The name of the fleet",
-        ).completer = FleetNameCompleter()
+        ).completer = FleetNameCompleter()  # type: ignore[attr-defined]
         delete_parser.add_argument(
             "-i",
             "--instance",

dstack/_internal/cli/commands/gateway.py

@@ -68,7 +68,7 @@ class GatewayCommand(APIBaseCommand):
         delete_parser.set_defaults(subfunc=self._delete)
         delete_parser.add_argument(
             "name", help="The name of the gateway"
-        ).completer = GatewayNameCompleter()
+        ).completer = GatewayNameCompleter()  # type: ignore[attr-defined]
         delete_parser.add_argument(
             "-y", "--yes", action="store_true", help="Don't ask for confirmation"
         )
@@ -79,7 +79,7 @@ class GatewayCommand(APIBaseCommand):
         update_parser.set_defaults(subfunc=self._update)
         update_parser.add_argument(
             "name", help="The name of the gateway"
-        ).completer = GatewayNameCompleter()
+        ).completer = GatewayNameCompleter()  # type: ignore[attr-defined]
         update_parser.add_argument(
             "--set-default", action="store_true", help="Set it the default gateway for the project"
         )

dstack/_internal/cli/commands/init.py

@@ -1,12 +1,17 @@
 import argparse
 import os
 from pathlib import Path
+from typing import Optional
 
 from dstack._internal.cli.commands import BaseCommand
-from dstack._internal.cli.services.repos import init_repo, register_init_repo_args
+from dstack._internal.cli.services.repos import (
+    get_repo_from_dir,
+    get_repo_from_url,
+    is_git_repo_url,
+    register_init_repo_args,
+)
 from dstack._internal.cli.utils.common import configure_logging, confirm_ask, console, warn
 from dstack._internal.core.errors import ConfigurationError
-from dstack._internal.core.models.repos.base import RepoType
 from dstack._internal.core.services.configs import ConfigManager
 from dstack.api import Client
 
@@ -21,6 +26,15 @@ class InitCommand(BaseCommand):
             help="The name of the project",
             default=os.getenv("DSTACK_PROJECT"),
         )
+        self._parser.add_argument(
+            "-P",
+            "--repo",
+            help=(
+                "The repo to initialize. Can be a local path or a Git repo URL."
+                " Defaults to the current working directory."
+            ),
+            dest="repo",
+        )
         register_init_repo_args(self._parser)
         # Deprecated since 0.19.25, ignored
         self._parser.add_argument(
@@ -30,7 +44,7 @@ class InitCommand(BaseCommand):
             type=Path,
             dest="ssh_identity_file",
         )
-        # A hidden mode for transitional period only, remove it with local repos
+        # A hidden mode for transitional period only, remove it with repos in `config.yml`
         self._parser.add_argument(
             "--remove",
             help=argparse.SUPPRESS,
@@ -39,44 +53,62 @@ class InitCommand(BaseCommand):
 
     def _command(self, args: argparse.Namespace):
         configure_logging()
+
+        repo_path: Optional[Path] = None
+        repo_url: Optional[str] = None
+        repo_arg: Optional[str] = args.repo
+        if repo_arg is not None:
+            if is_git_repo_url(repo_arg):
+                repo_url = repo_arg
+            else:
+                repo_path = Path(repo_arg).expanduser().resolve()
+        else:
+            repo_path = Path.cwd()
+
         if args.remove:
+            if repo_url is not None:
+                raise ConfigurationError(f"Local path expected, got URL: {repo_url}")
+            assert repo_path is not None
             config_manager = ConfigManager()
-            repo_path = Path.cwd()
             repo_config = config_manager.get_repo_config(repo_path)
             if repo_config is None:
-                raise ConfigurationError("The repo is not initialized, nothing to remove")
-            if repo_config.repo_type != RepoType.LOCAL:
-                raise ConfigurationError("`dstack init --remove` is for local repos only")
+                raise ConfigurationError("Repo record not found, nothing to remove")
             console.print(
-                f"You are about to remove the local repo {repo_path}\n"
+                f"You are about to remove the repo {repo_path}\n"
                 "Only the record about the repo will be removed,"
                 " the repo files will remain intact\n"
             )
-            if not confirm_ask("Remove the local repo?"):
+            if not confirm_ask("Remove the repo?"):
                 return
             config_manager.delete_repo_config(repo_config.repo_id)
             config_manager.save()
-            console.print("Local repo has been removed")
+            console.print("Repo has been removed")
             return
-        api = Client.from_config(
-            project_name=args.project, ssh_identity_file=args.ssh_identity_file
-        )
-        if args.local:
+
+        local: bool = args.local
+        if local:
             warn(
-                "Local repos are deprecated since 0.19.25 and will be removed soon."
-                " Consider using `files` instead: https://dstack.ai/docs/concepts/tasks/#files"
+                "Local repos are deprecated since 0.19.25 and will be removed soon. Consider"
+                " using [code]files[/code] instead: https://dstack.ai/docs/concepts/tasks/#files"
             )
         if args.ssh_identity_file:
             warn(
-                "`--ssh-identity` in `dstack init` is deprecated and ignored since 0.19.25."
-                " Use this option with `dstack apply` and `dstack attach` instead"
+                "[code]--ssh-identity[/code] in [code]dstack init[/code] is deprecated and ignored"
+                " since 0.19.25. Use this option with [code]dstack apply[/code]"
+                " and [code]dstack attach[/code] instead"
             )
-        init_repo(
-            api=api,
-            repo_path=Path.cwd(),
-            repo_branch=None,
-            repo_hash=None,
-            local=args.local,
+
+        if repo_url is not None:
+            # Dummy repo branch to avoid autodetection that fails on private repos.
+            # We don't need branch/hash for repo_id anyway.
+            repo = get_repo_from_url(repo_url, repo_branch="master")
+        elif repo_path is not None:
+            repo = get_repo_from_dir(repo_path, local=local)
+        else:
+            assert False, "should not reach here"
+        api = Client.from_config(project_name=args.project)
+        api.repos.init(
+            repo=repo,
             git_identity_file=args.git_identity_file,
             oauth_token=args.gh_token,
         )

dstack/_internal/cli/commands/logs.py

@@ -30,7 +30,7 @@ class LogsCommand(APIBaseCommand):
             type=int,
             default=0,
         )
-        self._parser.add_argument("run_name").completer = RunNameCompleter(all=True)
+        self._parser.add_argument("run_name").completer = RunNameCompleter(all=True)  # type: ignore[attr-defined]
 
     def _command(self, args: argparse.Namespace):
         super()._command(args)

dstack/_internal/cli/commands/metrics.py

@@ -26,7 +26,7 @@ class MetricsCommand(APIBaseCommand):
 
     def _register(self):
         super()._register()
-        self._parser.add_argument("run_name").completer = RunNameCompleter()
+        self._parser.add_argument("run_name").completer = RunNameCompleter()  # type: ignore[attr-defined]
         self._parser.add_argument(
             "-w",
             "--watch",

dstack/_internal/cli/commands/offer.py

@@ -3,7 +3,11 @@ from pathlib import Path
 from typing import List
 
 from dstack._internal.cli.commands import APIBaseCommand
-from dstack._internal.cli.services.configurators.run import BaseRunConfigurator
+from dstack._internal.cli.services.args import cpu_spec, disk_spec, gpu_spec
+from dstack._internal.cli.services.configurators.run import (
+    BaseRunConfigurator,
+)
+from dstack._internal.cli.services.profile import register_profile_args
 from dstack._internal.cli.utils.common import console
 from dstack._internal.cli.utils.gpu import print_gpu_json, print_gpu_table
 from dstack._internal.cli.utils.run import print_offers_json, print_run_plan
@@ -18,11 +22,8 @@ class OfferConfigurator(BaseRunConfigurator):
     TYPE = ApplyConfigurationType.TASK
 
     @classmethod
-    def register_args(
-        cls,
-        parser: argparse.ArgumentParser,
-    ):
-        super().register_args(parser, default_max_offers=50)
+    def register_args(cls, parser: argparse.ArgumentParser):
+        configuration_group = parser.add_argument_group(f"{cls.TYPE.value} Options")
         parser.add_argument(
             "--group-by",
             action="append",
@@ -33,6 +34,43 @@ class OfferConfigurator(BaseRunConfigurator):
                 "Can be repeated or comma-separated (e.g. [code]--group-by gpu,backend[/code])."
             ),
         )
+        configuration_group.add_argument(
+            "-n",
+            "--name",
+            dest="run_name",
+            help="The name of the run. If not specified, a random name is assigned",
+        )
+        configuration_group.add_argument(
+            "--max-offers",
+            help="Number of offers to show in the run plan",
+            type=int,
+            default=50,
+        )
+        cls.register_env_args(configuration_group)
+        configuration_group.add_argument(
+            "--cpu",
+            type=cpu_spec,
+            help="Request CPU for the run. "
+            "The format is [code]ARCH[/]:[code]COUNT[/] (all parts are optional)",
+            dest="cpu_spec",
+            metavar="SPEC",
+        )
+        configuration_group.add_argument(
+            "--gpu",
+            type=gpu_spec,
+            help="Request GPU for the run. "
+            "The format is [code]NAME[/]:[code]COUNT[/]:[code]MEMORY[/] (all parts are optional)",
+            dest="gpu_spec",
+            metavar="SPEC",
+        )
+        configuration_group.add_argument(
+            "--disk",
+            type=disk_spec,
+            help="Request the size range of disk for the run. Example [code]--disk 100GB..[/].",
+            metavar="RANGE",
+            dest="disk_spec",
+        )
+        register_profile_args(parser)
 
 
 class OfferCommand(APIBaseCommand):
@@ -117,7 +155,7 @@ class OfferCommand(APIBaseCommand):
 
         return processed
 
-    def _list_gpus(self, args: List[str], run_spec: RunSpec) -> List[GpuGroup]:
+    def _list_gpus(self, args: argparse.Namespace, run_spec: RunSpec) -> List[GpuGroup]:
         group_by = [g for g in args.group_by if g != "gpu"] or None
         return self.api.client.gpus.list_gpus(
             self.api.project,

dstack/_internal/cli/commands/project.py

@@ -124,8 +124,8 @@ class ProjectCommand(BaseCommand):
         table.add_column("USER", style="grey58")
         table.add_column("DEFAULT", justify="center")
 
-        for project_name in config_manager.list_projects():
-            project_config = config_manager.get_project_config(project_name)
+        for project_config in config_manager.list_project_configs():
+            project_name = project_config.name
             is_default = project_name == default_project.name if default_project else False
 
             # Get username from API

dstack/_internal/cli/commands/secrets.py

@@ -29,7 +29,7 @@ class SecretCommand(APIBaseCommand):
         get_parser.add_argument(
             "name",
             help="The name of the secret",
-        ).completer = SecretNameCompleter()
+        ).completer = SecretNameCompleter()  # type: ignore[attr-defined]
         get_parser.set_defaults(subfunc=self._get)
 
         set_parser = subparsers.add_parser(
@@ -53,7 +53,7 @@ class SecretCommand(APIBaseCommand):
         delete_parser.add_argument(
             "name",
             help="The name of the secret",
-        ).completer = SecretNameCompleter()
+        ).completer = SecretNameCompleter()  # type: ignore[attr-defined]
         delete_parser.add_argument(
             "-y", "--yes", help="Don't ask for confirmation", action="store_true"
         )

dstack/_internal/cli/commands/server.py

@@ -74,7 +74,7 @@ class ServerCommand(BaseCommand):
         uvicorn_log_level = os.getenv("DSTACK_SERVER_UVICORN_LOG_LEVEL", "ERROR").lower()
         reload_disabled = os.getenv("DSTACK_SERVER_RELOAD_DISABLED") is not None
 
-        uvicorn.run(
+        uvicorn.run(  # type: ignore[unbound-variable]
             "dstack._internal.server.main:app",
             host=args.host,
             port=args.port,

dstack/_internal/cli/commands/stop.py

@@ -14,7 +14,7 @@ class StopCommand(APIBaseCommand):
         super()._register()
         self._parser.add_argument("-x", "--abort", action="store_true")
         self._parser.add_argument("-y", "--yes", action="store_true")
-        self._parser.add_argument("run_name").completer = RunNameCompleter()
+        self._parser.add_argument("run_name").completer = RunNameCompleter()  # type: ignore[attr-defined]
 
     def _command(self, args: argparse.Namespace):
         super()._command(args)

dstack/_internal/cli/commands/volume.py

@@ -48,7 +48,7 @@ class VolumeCommand(APIBaseCommand):
         delete_parser.add_argument(
             "name",
             help="The name of the volume",
-        ).completer = VolumeNameCompleter()
+        ).completer = VolumeNameCompleter()  # type: ignore[attr-defined]
         delete_parser.add_argument(
             "-y", "--yes", help="Don't ask for confirmation", action="store_true"
         )

dstack/_internal/cli/main.py

@@ -41,8 +41,8 @@ def main():
 
     parser = argparse.ArgumentParser(
         description=(
-            "Not sure where to start? Call [code]dstack init[/].\n"
-            "Define a [code].dstack.yml[/] configuration file and run it via [code]dstack apply[/]\n"
+            "Not sure where to start?"
+            " Define a [code].dstack.yml[/] configuration file and run it via [code]dstack apply[/]\n"
         ),
         formatter_class=RichHelpFormatter,
         epilog=(

dstack/_internal/cli/services/completion.py

@@ -87,5 +87,5 @@ class ProjectNameCompleter(BaseCompleter):
 
     def __call__(self, prefix: str, parsed_args: argparse.Namespace, **kwargs) -> List[str]:
         argcomplete.debug(f"{self.__class__.__name__}: Listing projects from ConfigManager")
-        projects = ConfigManager().list_projects()
-        return [p for p in projects if p.startswith(prefix)]
+        projects = ConfigManager().list_project_configs()
+        return [p.name for p in projects if p.name.startswith(prefix)]

dstack/_internal/cli/services/configurators/__init__.py

@@ -24,7 +24,9 @@ from dstack._internal.core.models.configurations import (
 APPLY_STDIN_NAME = "-"
 
 
-apply_configurators_mapping: Dict[ApplyConfigurationType, Type[BaseApplyConfigurator]] = {
+apply_configurators_mapping: Dict[
+    ApplyConfigurationType, Type[BaseApplyConfigurator[AnyApplyConfiguration]]
+] = {
     cls.TYPE: cls
     for cls in [
         DevEnvironmentConfigurator,
@@ -47,7 +49,9 @@ run_configurators_mapping: Dict[ApplyConfigurationType, Type[BaseRunConfigurator
 }
 
 
-def get_apply_configurator_class(configurator_type: str) -> Type[BaseApplyConfigurator]:
+def get_apply_configurator_class(
+    configurator_type: str,
+) -> Type[BaseApplyConfigurator[AnyApplyConfiguration]]:
     return apply_configurators_mapping[ApplyConfigurationType(configurator_type)]
 
 

dstack/_internal/cli/services/configurators/base.py

@@ -1,7 +1,7 @@
 import argparse
 import os
 from abc import ABC, abstractmethod
-from typing import List, Optional, Union, cast
+from typing import Generic, List, TypeVar, Union, cast
 
 from dstack._internal.cli.services.args import env_var
 from dstack._internal.core.errors import ConfigurationError
@@ -10,13 +10,14 @@ from dstack._internal.core.models.configurations import (
     ApplyConfigurationType,
 )
 from dstack._internal.core.models.envs import Env, EnvSentinel, EnvVarTuple
-from dstack._internal.core.models.repos.base import Repo
 from dstack.api._public import Client
 
 ArgsParser = Union[argparse._ArgumentGroup, argparse.ArgumentParser]
 
+ApplyConfigurationT = TypeVar("ApplyConfigurationT", bound=AnyApplyConfiguration)
 
-class BaseApplyConfigurator(ABC):
+
+class BaseApplyConfigurator(ABC, Generic[ApplyConfigurationT]):
     TYPE: ApplyConfigurationType
 
     def __init__(self, api_client: Client):
@@ -25,12 +26,11 @@ class BaseApplyConfigurator(ABC):
     @abstractmethod
     def apply_configuration(
         self,
-        conf: AnyApplyConfiguration,
+        conf: ApplyConfigurationT,
         configuration_path: str,
         command_args: argparse.Namespace,
         configurator_args: argparse.Namespace,
         unknown_args: List[str],
-        repo: Optional[Repo] = None,
     ):
         """
         Implements `dstack apply` for a given configuration type.
@@ -41,14 +41,13 @@ class BaseApplyConfigurator(ABC):
             command_args: The args parsed by `dstack apply`.
             configurator_args: The known args parsed by `cls.get_parser()`.
             unknown_args: The unknown args after parsing by `cls.get_parser()`.
-            repo: The repo to use with apply.
         """
         pass
 
     @abstractmethod
     def delete_configuration(
         self,
-        conf: AnyApplyConfiguration,
+        conf: ApplyConfigurationT,
         configuration_path: str,
         command_args: argparse.Namespace,
     ):

dstack/_internal/cli/services/configurators/fleet.py

@@ -35,7 +35,6 @@ from dstack._internal.core.models.fleets import (
     InstanceGroupPlacement,
 )
 from dstack._internal.core.models.instances import InstanceAvailability, InstanceStatus, SSHKey
-from dstack._internal.core.models.repos.base import Repo
 from dstack._internal.core.services.diff import diff_models
 from dstack._internal.utils.common import local_time
 from dstack._internal.utils.logging import get_logger
@@ -46,7 +45,7 @@ from dstack.api.utils import load_profile
 logger = get_logger(__name__)
 
 
-class FleetConfigurator(ApplyEnvVarsConfiguratorMixin, BaseApplyConfigurator):
+class FleetConfigurator(ApplyEnvVarsConfiguratorMixin, BaseApplyConfigurator[FleetConfiguration]):
     TYPE: ApplyConfigurationType = ApplyConfigurationType.FLEET
 
     def apply_configuration(
@@ -56,7 +55,6 @@ class FleetConfigurator(ApplyEnvVarsConfiguratorMixin, BaseApplyConfigurator):
         command_args: argparse.Namespace,
         configurator_args: argparse.Namespace,
         unknown_args: List[str],
-        repo: Optional[Repo] = None,
     ):
         self.apply_args(conf, configurator_args, unknown_args)
         profile = load_profile(Path.cwd(), None)

dstack/_internal/cli/services/configurators/gateway.py

@@ -1,6 +1,6 @@
 import argparse
 import time
-from typing import List, Optional
+from typing import List
 
 from rich.table import Table
 
@@ -21,13 +21,12 @@ from dstack._internal.core.models.gateways import (
     GatewaySpec,
     GatewayStatus,
 )
-from dstack._internal.core.models.repos.base import Repo
 from dstack._internal.core.services.diff import diff_models
 from dstack._internal.utils.common import local_time
 from dstack.api._public import Client
 
 
-class GatewayConfigurator(BaseApplyConfigurator):
+class GatewayConfigurator(BaseApplyConfigurator[GatewayConfiguration]):
     TYPE: ApplyConfigurationType = ApplyConfigurationType.GATEWAY
 
     def apply_configuration(
@@ -37,7 +36,6 @@ class GatewayConfigurator(BaseApplyConfigurator):
         command_args: argparse.Namespace,
         configurator_args: argparse.Namespace,
         unknown_args: List[str],
-        repo: Optional[Repo] = None,
     ):
         self.apply_args(conf, configurator_args, unknown_args)
         spec = GatewaySpec(