Support calling RunCollection.get_plan() without repo (#1741)

dstackai · Sep 30, 2024 · fccc8dd · fccc8dd
1 parent ec8d4a7
commit fccc8dd
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/dstack/api/_public/runs.py b/src/dstack/api/_public/runs.py
@@ -418,7 +418,7 @@ def create_instance(self, profile: Profile, requirements: Requirements) -> Insta
     def get_plan(
         self,
         configuration: AnyRunConfiguration,
-        repo: Repo,
+        repo: Optional[Repo] = None,
         configuration_path: Optional[str] = None,
         backends: Optional[List[BackendType]] = None,
         regions: Optional[List[str]] = None,
@@ -442,6 +442,12 @@ def get_plan(
         # Returns:
         #     run plan
         # """
+
+        if repo is None:
+            repo = configuration.get_repo()
+            if repo is None:
+                raise ConfigurationError("Repo is required for this type of configuration")
+
         if working_dir is None:
             working_dir = "."
         elif repo.repo_dir is not None: