about summary refs log tree commit diff
path: root/wqflask/utility/helper_functions.py
diff options
context:
space:
mode:
Diffstat (limited to 'wqflask/utility/helper_functions.py')
-rw-r--r--wqflask/utility/helper_functions.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/wqflask/utility/helper_functions.py b/wqflask/utility/helper_functions.py
index 920d9ac6..d2567b63 100644
--- a/wqflask/utility/helper_functions.py
+++ b/wqflask/utility/helper_functions.py
@@ -2,14 +2,16 @@ from __future__ import absolute_import, print_function, division
 
 from base.trait import GeneralTrait
 from base import data_set
+from base.species import TheSpecies
 
-def get_dataset_and_trait(self, start_vars):
+def get_species_dataset_trait(self, start_vars):
     #assert type(read_genotype) == type(bool()), "Expecting boolean value for read_genotype"
     self.dataset = data_set.create_dataset(start_vars['dataset'])
+    self.species = TheSpecies(dataset=self.dataset)
     self.this_trait = GeneralTrait(dataset=self.dataset.name,
                                    name=start_vars['trait_id'],
                                    cellid=None)
     
     #if read_genotype:
     self.dataset.group.read_genotype_file()
-    self.genotype = self.dataset.group.genotype    
\ No newline at end of file
+    self.genotype = self.dataset.group.genotype