diff --git a/workflows/smokers.py b/workflows/smokers.py index 558e7f7..03c6563 100644 --- a/workflows/smokers.py +++ b/workflows/smokers.py @@ -1,4 +1,5 @@ import sys,os +import os.path import webbrowser import cPickle @@ -76,7 +77,7 @@ class DatasetLoadFunctionSmokerSmall(workflow.Function): workflow.Function.__init__(self, 'load_small', 'Smoker (Small)') def run(self): - path = 'data/smokers-small/' + path = os.path.join(main.options.datadir, 'smokers-small/') files = os.listdir(path) out = [] for fname in files: @@ -92,7 +93,7 @@ class DatasetLoadFunctionSmokerMedium(workflow.Function): workflow.Function.__init__(self, 'load_medium', 'Smoker (Medium)') def run(self): - path = 'data/smokers-medium/' + path = os.path.join(main.options.datadir, 'smokers-medium/') files = os.listdir(path) out = [] for fname in files: @@ -108,7 +109,7 @@ class DatasetLoadFunctionSmokerFull(workflow.Function): workflow.Function.__init__(self, 'load_full', 'Smoker (Full)') def run(self): - path = 'data/smokers-full/' + path = os.path.join(main.options.datadir, 'smokers-full/') files = os.listdir(path) out = [] for fname in files: @@ -123,7 +124,7 @@ class DatasetLoadFunctionSmokerGO(workflow.Function): workflow.Function.__init__(self, 'load_go', 'Smoker (GO)') def run(self): - path = 'data/smokers-go/' + path = os.path.join(main.options.datadir, 'smokers-go/') files = os.listdir(path) out = [] for fname in files: