Merge pull request #715 from edmundcrawley/master

Bug fix to drawDiscrete
econ-ark · Jun 17, 2020 · d93e80c · d93e80c
2 parents a55d06f + 3f69148
commit d93e80c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/HARK/ConsumptionSaving/tests/test_ConsAggShockModel.py b/HARK/ConsumptionSaving/tests/test_ConsAggShockModel.py
@@ -79,5 +79,5 @@ def test_economy(self):
 
         self.economy.AFunc = self.economy.dynamics.AFunc
         self.assertAlmostEqual(self.economy.AFunc[0].slope,
-                         1.0921217053006234)
+                         1.0845554708377696)
 
diff --git a/HARK/distribution.py b/HARK/distribution.py
@@ -514,9 +514,9 @@ def drawDiscrete(self, N,X=None,exact_match=False,seed=0):
                 bot = top
                 top = cutoffs[j]
                 event_list += (top-bot)*[events[j]]
-                # Randomly permute the event indices and store the corresponding results
-                event_draws = RNG.permutation(event_list)
-                draws = X[event_draws]
+            # Randomly permute the event indices and store the corresponding results
+            event_draws = RNG.permutation(event_list)
+            draws = X[event_draws]
         else:
             indices = self.draw_events(N, seed=seed)
             if J > 1: