Fixes minor numpy.eye error that was thrown for unstable learned mo…

…dels.

ChangeLog.md

@@ -1,6 +1,8 @@
 # Changes 
 Versioning follows [semver](https://semver.org/).
 
+- v1.2.5:
+  - Fixes minor `numpy.eye` error that was thrown for unstable learned models.
 - v1.2.0:
   - Adds version with support for external input (i.e., IPSID).
 - v1.1.0:

setup.py

@@ -27,7 +27,7 @@
 
 setuptools.setup(
     name="PSID",
-    version="1.2.4",
+    version="1.2.5",
     author="Omid Sani",
     author_email="omidsani@gmail.com",
     description="Python implementation for preferential subspace identification (PSID)",

source/PSID/LSSM.py

@@ -143,8 +143,8 @@ def update_secondary_params(self):
                 self.YCov = self.C @ self.XCov @ self.C.T + self.R
                 self.YCov = (self.YCov + self.YCov.T)/2
             else:
-                self.XCov = np.eye(self.A.shape); self.XCov[:] = np.nan
-                self.YCov = np.eye(self.C.shape); self.YCov[:] = np.nan
+                self.XCov = np.eye(self.state_dim); self.XCov[:] = np.nan
+                self.YCov = np.eye(self.output_dim); self.YCov[:] = np.nan
 
             try:
                 self.Pp = linalg.solve_discrete_are(self.A.T, self.C.T, self.Q, self.R, s=self.S) # Solves Katayama eq. 5.42a