Temporary fix for non-deterministic tests

JohnVinyard · Aug 25, 2017 · f582cb1 · f582cb1
1 parent 4d4296c
commit f582cb1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/zounds/learn/test_learn.py b/zounds/learn/test_learn.py
@@ -46,7 +46,7 @@ class Rbm(featureflow.BaseModel, Settings):
 
         rbm = featureflow.PickleFeature(
             KMeans,
-            centroids=64,
+            centroids=3,
             needs=meanstd,
             store=False)
 
@@ -76,7 +76,7 @@ def test_can_use_learned_feature(self):
         KMeans.process(iterator=data())
         l = Learned(learned=KMeans())
         results = list(l._process(np.random.random_sample((33, 3))))[0]
-        self.assertEqual((33, 64), results.shape)
+        self.assertEqual((33, 3), results.shape)
 
     def test_pipeline_changes_version_when_recomputed(self):
         KMeans = build_classes()