From 298db0fb1bf8506d4129510c04dd4fe7a1147fe8 Mon Sep 17 00:00:00 2001
From: Gregor Sturm <mail@gregor-sturm.de>
Date: Thu, 26 Dec 2024 14:10:32 +0100
Subject: [PATCH] Remove contrast dataclass, which isnt used anywhere

---
 .../tools/_differential_gene_expression/__init__.py  |  3 +--
 pertpy/tools/_differential_gene_expression/_base.py  | 12 ------------
 2 files changed, 1 insertion(+), 14 deletions(-)

diff --git a/pertpy/tools/_differential_gene_expression/__init__.py b/pertpy/tools/_differential_gene_expression/__init__.py
index 35ccef51..6cc925dd 100644
--- a/pertpy/tools/_differential_gene_expression/__init__.py
+++ b/pertpy/tools/_differential_gene_expression/__init__.py
@@ -1,4 +1,4 @@
-from ._base import ContrastType, LinearModelBase, MethodBase
+from ._base import LinearModelBase, MethodBase
 from ._dge_comparison import DGEEVAL
 from ._edger import EdgeR
 from ._pydeseq2 import PyDESeq2
@@ -14,7 +14,6 @@
     "SimpleComparisonBase",
     "WilcoxonTest",
     "TTest",
-    "ContrastType",
 ]
 
 AVAILABLE_METHODS = [Statsmodels, EdgeR, PyDESeq2, WilcoxonTest, TTest]
diff --git a/pertpy/tools/_differential_gene_expression/_base.py b/pertpy/tools/_differential_gene_expression/_base.py
index 4cd0dda0..011a5630 100644
--- a/pertpy/tools/_differential_gene_expression/_base.py
+++ b/pertpy/tools/_differential_gene_expression/_base.py
@@ -25,18 +25,6 @@
 from pertpy.tools._differential_gene_expression._checks import check_is_numeric_matrix
 
 
-@dataclass
-class Contrast:
-    """Simple contrast for comparison between groups"""
-
-    column: str
-    baseline: str
-    group_to_compare: str
-
-
-ContrastType = Contrast | tuple[str, str, str]
-
-
 class MethodBase(ABC):
     def __init__(self, adata, *, mask=None, layer=None, **kwargs):
         """