Fix bug in retrieving samples (#70)

* fix bug in retrieving samples * resolve PR reviews --------- Co-authored-by: Apostolos Chalkis <[email protected]>
GeomScale · Jul 18, 2023 · f5440b6 · f5440b6
1 parent c71b3c6
commit f5440b6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/dingo/PolytopeSampler.py b/dingo/PolytopeSampler.py
@@ -203,9 +203,10 @@ def generate_steady_states_no_multiphase(
         P = HPolytope(self._A, self._b)
 
         samples = P.generate_samples(method, n, burn_in, thinning, self._parameters["fast_computations"])
+        samples_T = samples.T
 
         steady_states = map_samples_to_steady_states(
-                samples, self._N, self._N_shift
+                samples_T, self._N, self._N_shift
             )
 
         return steady_states
@@ -263,7 +264,8 @@ def sample_from_polytope_no_multiphase(
         except ImportError as e:
             samples = P.generate_samples(method, n, burn_in, thinning, False)
 
-        return samples
+        samples_T = samples.T
+        return samples_T
 
     @staticmethod
     def round_polytope(