[thermo, subset] fixing ThermoMSM

markovmodel · Sep 6, 2016 · 3ee3472 · 3ee3472
1 parent cc1a754
commit 3ee3472
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/pyemma/thermo/models/memm.py b/pyemma/thermo/models/memm.py
@@ -97,6 +97,15 @@ def stationary_distribution(self):
     @property
     def pi_full_state(self):
         return self.stationary_distribution_full_state
+
+    @_map_to_full_state(default_arg=_np.inf)
+    def eigenvectors_right(self, k=None):
+        return super(ThermoMSM, self).eigenvectors_right(k=k)
+
+    @_map_to_full_state(default_arg=_np.inf, extend_along_axis=1)
+    def eigenvectors_left(self, k=None):
+        return super(ThermoMSM, self).eigenvectors_left(k=k)
+
 
 
 class MEMM(_MultiThermModel):