From ebf288807e80a29779eefa6e399e8bea64d1d6d1 Mon Sep 17 00:00:00 2001
From: Marius <44572437+m-a-r-i-u-s@users.noreply.github.com>
Date: Mon, 16 Sep 2024 14:30:41 +0200
Subject: [PATCH] Update base.py

Added additional argument for chemical cluster groups
---
 domhmm/analysis/base.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/domhmm/analysis/base.py b/domhmm/analysis/base.py
index 92c94ec..c8e30e3 100644
--- a/domhmm/analysis/base.py
+++ b/domhmm/analysis/base.py
@@ -111,6 +111,7 @@ def __init__(
             sterol_heads: Dict[str, Any] = {},
             sterol_tails: Dict[str, Any] = {},
             tmd_protein_list: Union[None, "AtomGroup", str, list] = None,
+            cluster_group: list = [],
             frac: float = 0.5,
             p_value: float = 0.05,
             leaflet_frame_rate: Union[None, int] = None,
@@ -146,6 +147,7 @@ def __init__(
         self.verbose = verbose
         self.result_plots = result_plots
         self.tmd_protein = None
+        self.cluster_group = cluster_group
         self.n_init_hmm = n_init_hmm
 
         assert heads.keys() == tails.keys(), "Heads and tails don't contain same residue names"