Fix test_lpi_bounds to use correct cardinalities in set_distribution

Author: eliewolfe

test/test_pipeline.py

@@ -574,12 +574,12 @@ def test_lpi_bounds(self):
         sdp  = InflationSDP(trivial)
         cols = [np.array([]),
                 np.array([[1, 2, 0, 0],
-                 [1, 2, 1, 0]]),
+                          [1, 2, 1, 0]]),
                 np.array([[1, 1, 0, 0],
-                 [1, 2, 0, 0],
-                 [1, 2, 1, 0]])]
+                          [1, 2, 0, 0],
+                          [1, 2, 1, 0]])]
         sdp.generate_relaxation(cols)
-        sdp.set_distribution(np.ones((2, 1)) / 2,
+        sdp.set_distribution(np.ones((2, 2)) / 2,
                              use_lpi_constraints=True)
 
         self.assertGreaterEqual(len(sdp.semiknown_moments), 1,