removed remaining xarray references

Author: davidsebfischer

diffxpy/models/batch_bfgs/optim.py

@@ -6,7 +6,6 @@
 from numpy.linalg import pinv
 import numpy.random
 from multiprocessing import Pool
-import xarray as xr
 
 from .objectives import *
 

diffxpy/models/hessian.py

@@ -27,9 +27,9 @@ def hes_nb_glm_mean_block(
         Estimated mean parameters across cells for a given gene.
     :param mu: np.ndarray (cells,)
         Estimated dispersion parameters across cells for a given gene.
-    :param design_loc: np.ndarray, matrix, xarray (cells, #parameters location model)
+    :param design_loc: np.ndarray, matrix (cells, #parameters location model)
         Design matrix of location model.
-    :param design_scale: np.ndarray, matrix, xarray (cells, #parameters shape model)
+    :param design_scale: np.ndarray, matrix (cells, #parameters shape model)
         Design matrix of shape model.
     :param i: int
         Index of first dimension in fisher information matrix which is to be computed.
@@ -78,9 +78,9 @@ def hes_nb_glm_disp_block(
         Estimated mean parameters across cells for a given gene.
     :param mu: np.ndarray (cells,)
         Estimated dispersion parameters across cells for a given gene.
-    :param design_loc: np.ndarray, matrix, xarray (cells, #parameters location model)
+    :param design_loc: np.ndarray, matrix (cells, #parameters location model)
         Design matrix of location model.
-    :param design_scale: np.ndarray, matrix, xarray (cells, #parameters shape model)
+    :param design_scale: np.ndarray, matrix (cells, #parameters shape model)
         Design matrix of shape model.
     :param i: int
         Index of first dimension in fisher information matrix which is to be computed.
@@ -129,9 +129,9 @@ def hes_nb_glm_meandisp_block(
         Estimated mean parameters across cells for a given gene.
     :param mu: np.ndarray (cells,)
         Estimated dispersion parameters across cells for a given gene.
-    :param design_loc: np.ndarray, matrix, xarray (cells, #parameters location model)
+    :param design_loc: np.ndarray, matrix (cells, #parameters location model)
         Design matrix of location model.
-    :param design_scale: np.ndarray, matrix, xarray (cells, #parameters shape model)
+    :param design_scale: np.ndarray, matrix (cells, #parameters shape model)
         Design matrix of shape model.
     :param i: int
         Index of first dimension in fisher information matrix which is to be computed.
@@ -160,9 +160,9 @@ def hes_nb_glm_bygene(
         Estimated mean parameters across cells for a given gene.
     :param mu: np.ndarray (cells,)
         Estimated dispersion parameters across cells for a given gene.
-    :param design_loc: np.ndarray, matrix, xarray (cells, #parameters location model)
+    :param design_loc: np.ndarray, matrix (cells, #parameters location model)
         Design matrix of location model.
-    :param design_scale: np.ndarray, matrix, xarray (cells, #parameters shape model)
+    :param design_scale: np.ndarray, matrix (cells, #parameters shape model)
         Design matrix of shape model.
     
     :return: np.ndarray (#parameters location model + #parameters shape model, #parameters location model + #parameters shape model)
@@ -212,9 +212,9 @@ def theta_covar_bygene(
         Estimated mean parameters across cells for a given gene.
     :param mu: np.ndarray (cells,)
         Estimated dispersion parameters across cells for a given gene.
-    :param design_loc: np.ndarray, matrix, xarray (cells, #parameters location model)
+    :param design_loc: np.ndarray, matrix (cells, #parameters location model)
         Design matrix of location model.
-    :param design_scale: np.ndarray, matrix, xarray (cells, #parameters shape model)
+    :param design_scale: np.ndarray, matrix(cells, #parameters shape model)
         Design matrix of shape model.
     
     :return: np.ndarray (#parameters location model + #parameters shape model, #parameters location model + #parameters shape model)
@@ -243,9 +243,9 @@ def theta_sd_bygene(
         Estimated mean parameters across cells for a given gene.
     :param disp: np.ndarray (cells,)
         Estimated dispersion parameters across cells for a given gene.
-    :param design_loc: np.ndarray, matrix, xarray (cells, #parameters location model)
+    :param design_loc: np.ndarray, matrix (cells, #parameters location model)
         Design matrix of location model.
-    :param design_scale: np.ndarray, matrix, xarray (cells, #parameters shape model)
+    :param design_scale: np.ndarray, matrix (cells, #parameters shape model)
         Design matrix of shape model.
     
     :return: np.ndarray (#parameters location model + #parameters shape model,)

diffxpy/testing/det.py

@@ -581,10 +581,8 @@ def log_fold_change(self, base=np.e, return_type="vector"):
 
             - "dataframe":
               return a pandas.DataFrame with columns `gene`, `minuend_<group>`, `subtrahend_<group>` and `logFC`.
-            - "xarray":
-              return a xarray.DataArray with dimensions `(minuend, subtrahend, gene)`
 
-        :return: either pandas.DataFrame or xarray.DataArray
+        :return:
         """
         factors = set(self.full_design_loc_info.term_names) - set(self.reduced_design_loc_info.term_names)
 

diffxpy/testing/tests.py

@@ -594,7 +594,7 @@ def wald(
                 # there is no such column; modify design matrix to create one
                 design_loc[:, col_indices] = np.where(samples, 1, 0)
     elif coef_to_test is not None:
-        # Directly select coefficients to test from design matrix (xarray):
+        # Directly select coefficients to test from design matrix:
         # Check that coefficients to test are not dependent parameters if constraints are given:
         coef_loc_names = glm.data.view_coef_names(design_loc).tolist()
         if not np.all([x in coef_loc_names for x in coef_to_test]):

diffxpy/testing/utils.py

@@ -123,7 +123,7 @@ def design_matrix(
         formula: Union[None, str] = None,
         as_numeric: Union[List[str], Tuple[str], str] = (),
         dmat: Union[pd.DataFrame, None] = None,
-        return_type: str = "xarray"
+        return_type: str = "patsy"
 ) -> Union[patsy.design_info.DesignMatrix, pd.DataFrame]:
     """ Create a design matrix from some sample description.