Add a new option allowing client code to turn off parallelism

JohnVinyard · Jun 25, 2018 · 6a17674 · 6a17674
1 parent b4ab098
commit 6a17674
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/zounds/learn/graph.py b/zounds/learn/graph.py
@@ -29,7 +29,9 @@ class InfiniteLearningPipeline(cls):
         dataset = ff.Feature(
             InfiniteSampler,
             nsamples=ff.Var('nsamples'),
-            dtype=ff.Var('dtype'))
+            dtype=ff.Var('dtype'),
+            feature_filter=ff.Var('feature_filter'),
+            parallel=ff.Var('parallel'))
 
         pipeline = ff.ClobberPickleFeature(
             PreprocessingPipeline,