Merge pull request #40 from lsst-sqre/tickets/DM-29799

[DM-29799] Allow configuration of repo url and branch
lsst-sqre · Apr 22, 2021 · ee4ce22 · ee4ce22
2 parents be2f287 + 5e88e8f
commit ee4ce22
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/mobu/notebookrunner.py b/src/mobu/notebookrunner.py
@@ -48,8 +48,11 @@ async def run(self) -> None:
             )
 
             if not self._repo:
+                repo_url = self.options.get("repo_url", REPO_URL)
+                repo_branch = self.options.get("repo_branch", REPO_BRANCH)
+
                 self._repo = git.Repo.clone_from(
-                    REPO_URL, self._repo_dir.name, branch=REPO_BRANCH
+                    repo_url, self._repo_dir.name, branch=repo_branch
                 )
 
             self._notebook_iterator = os.scandir(self._repo_dir.name)