From f075b0b6f69f1bae3735eb8560b8f571c708aafd Mon Sep 17 00:00:00 2001
From: Sander Vrijders <sander.vrijders@ugent.be>
Date: Mon, 12 Feb 2018 18:36:07 +0100
Subject: prototypes: Add option to choose git repo and branch

This adds two extra parameters to the Experiment class, so that users
can specify their custom repositories and branches for testing new
features.
---
 rumba/model.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'rumba/model.py')

diff --git a/rumba/model.py b/rumba/model.py
index 14b856a..b877209 100644
--- a/rumba/model.py
+++ b/rumba/model.py
@@ -526,10 +526,12 @@ class Policy(object):
 class Experiment(object):
     __metaclass__ = abc.ABCMeta
 
-    def __init__(self, testbed, nodes=None):
+    def __init__(self, testbed, nodes=None, git_repo=None, git_branch=None):
         if nodes is None:
             nodes = list()
         self.nodes = nodes
+        self.git_repo = git_repo
+        self.git_branch = git_branch
         self.testbed = testbed
         # the strategy employed for completing the enrollment phase in
         # the different DIFs
-- 
cgit v1.2.3