undo old changes to _identity.py

Author: Jimmy-INL

pysensors/basis/_identity.py

@@ -53,8 +53,6 @@ def fit(self, X):
         -------
         self : instance
         """
-        # Store original data
-        self.original_data = X
         # Note that we take a transpose here, so columns correspond to examples
         if self.n_basis_modes is None:
             self.basis_matrix_ = check_array(X).T.copy()
@@ -67,10 +65,10 @@ def fit(self, X):
                     )
                 )
 
-            self.basis_matrix_ = np.eye(X.shape[1])[:,:self.n_basis_modes] #check_array(X)[: self.n_basis_modes, :].T.copy()
+            self.basis_matrix_ = check_array(X)[: self.n_basis_modes, :].T.copy() # np.eye(X.shape[1])[:,:self.n_basis_modes]
 
-            # if self.n_basis_modes < X.shape[0]:
-            #     warn(f"Only the first {self.n_basis_modes} examples were retained.")
+            if self.n_basis_modes < X.shape[0]:
+                warn(f"Only the first {self.n_basis_modes} examples were retained.")
         return self
 
     def matrix_inverse(self, n_basis_modes=None):