SURE-tools 2.1.36__py3-none-any.whl → 2.1.37__py3-none-any.whl

SURE/perturb/perturb.py

@@ -8,7 +8,7 @@ class LabelMatrix:
     def __init__(self):
         self.labels_ = None
     
-    def fit_transform(self, labels, control_label, sep_pattern=r'[;_\-\s]', speedup: Literal['none','vectorize','parallel']='none'):
+    def fit_transform(self, labels, control_label=None, sep_pattern=r'[;_\-\s]', speedup: Literal['none','vectorize','parallel']='none'):
         if speedup=='none':
             mat, self.labels_ = label_to_matrix(labels=labels, sep_pattern=sep_pattern)
         elif speedup=='vectorize':
@@ -18,9 +18,11 @@ class LabelMatrix:
             
         self.labels_ = np.array(self.labels_)
         
-        idx = np.where(self.labels_==control_label)[0]
-        mat = np.delete(mat, idx, axis=1)
-        self.labels_ = np.delete(self.labels_, idx)
+        if control_label is not None:
+            idx = np.where(self.labels_==control_label)[0]
+            mat = np.delete(mat, idx, axis=1)
+            self.labels_ = np.delete(self.labels_, idx)
+            
         return mat
             
     def inverse_transform(self, matrix):

{sure_tools-2.1.36.dist-info → sure_tools-2.1.37.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SURE-tools
-Version: 2.1.36
+Version: 2.1.37
 Summary: Succinct Representation of Single Cells
 Home-page: https://github.com/ZengFLab/SURE
 Author: Feng Zeng

{sure_tools-2.1.36.dist-info → sure_tools-2.1.37.dist-info}/RECORD

@@ -12,14 +12,14 @@ SURE/flow/__init__.py,sha256=rsAjYsh1xVIrxBCuwOE0Q_6N5th1wBgjJceV0ABPG3c,183
 SURE/flow/flow_stats.py,sha256=cBBsPEDpWNMpbzlyQ3f0385RSrX6_5RCH2caOyi4ihM,9908
 SURE/flow/plot_quiver.py,sha256=UbmuScUcgbQHeMmjKmgqxjrIjHhiHx0VWct16UMMwuE,8110
 SURE/perturb/__init__.py,sha256=ouxShhbxZM4r5Gf7GmKiutrsmtyq7QL8rHjhgF0BU08,32
-SURE/perturb/perturb.py,sha256=GVNgfNLGKic_jOwuhMLW8VZa2Md_9O3cwL2Dwik-sTI,3476
+SURE/perturb/perturb.py,sha256=uUFUNJzIzvjA_2rN8801Le-i3oS7vOEvFav1w_lSFSE,3544
 SURE/utils/__init__.py,sha256=QJUOfrXzdWSmoM0P3LH8oKEHttzCWqpDy2UF0F0dtN4,673
 SURE/utils/custom_mlp.py,sha256=rHnx9jEef02zfCUdbYVCmbuHcDdIBmRgt__wpdpZvYg,8104
 SURE/utils/queue.py,sha256=E_5PA5EWcBoGAZj8BkKQnkCK0p4C-4-xcTPqdIXaPXU,1892
 SURE/utils/utils.py,sha256=IUHjDDtYaAYllCWsZyIzqQwaLul6fJRvHRH4vIYcR-c,8462
-sure_tools-2.1.36.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
-sure_tools-2.1.36.dist-info/METADATA,sha256=ImWAYH50tlvLpd6rDzNya96MEQbICgIzbq2I1mcsAzY,2651
-sure_tools-2.1.36.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sure_tools-2.1.36.dist-info/entry_points.txt,sha256=-nJI8rVe_qqrR0HmfAODzj-JNfEqCcSsyVh6okSqyHk,83
-sure_tools-2.1.36.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
-sure_tools-2.1.36.dist-info/RECORD,,
+sure_tools-2.1.37.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
+sure_tools-2.1.37.dist-info/METADATA,sha256=9OEHt9VYrCralJ-l5Ssbk8lAie-8BGHRvwNNpUUqG_E,2651
+sure_tools-2.1.37.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sure_tools-2.1.37.dist-info/entry_points.txt,sha256=-nJI8rVe_qqrR0HmfAODzj-JNfEqCcSsyVh6okSqyHk,83
+sure_tools-2.1.37.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
+sure_tools-2.1.37.dist-info/RECORD,,