From 92b3d31682294a0f337b157f96ae3efdea9726a7 Mon Sep 17 00:00:00 2001
From: Daniel Wiesmann <daniel@wiesmann.pt>
Date: Wed, 31 Jan 2024 18:32:05 +0000
Subject: [PATCH] Remove stale bands argument from clay model classes

---
 src/model_clay.py | 7 -------
 1 file changed, 7 deletions(-)

diff --git a/src/model_clay.py b/src/model_clay.py
index 8f1d1eb4..cffea57e 100644
--- a/src/model_clay.py
+++ b/src/model_clay.py
@@ -62,7 +62,6 @@ def __init__(  # noqa: PLR0913
         heads,
         dim_head,
         mlp_ratio,
-        bands,
         band_groups,
         dropout,
         emb_dropout,
@@ -75,7 +74,6 @@ def __init__(  # noqa: PLR0913
         self.image_size = image_size
         self.patch_size = patch_size
         self.dim = dim
-        self.bands = bands
         self.band_groups = band_groups
         self.num_spatial_patches = (image_size // patch_size) ** 2
         self.num_group_patches = len(band_groups)
@@ -352,7 +350,6 @@ def __init__(  # noqa: PLR0913
         heads,
         dim_head,
         mlp_ratio,
-        bands,
         band_groups,
         dropout,
     ):
@@ -579,7 +576,6 @@ def __init__(  # noqa: PLR0913
         decoder_mlp_ratio,
         decoder_dropout,
         # EO
-        bands=13,
         band_groups={
             "rgb": (2, 1, 0),
             "rededge": (3, 4, 5, 7),
@@ -594,7 +590,6 @@ def __init__(  # noqa: PLR0913
         self.mask_ratio = mask_ratio
         self.image_size = image_size
         self.patch_size = patch_size
-        self.bands = bands
         self.band_groups = band_groups
 
         self.encoder = Encoder(
@@ -606,7 +601,6 @@ def __init__(  # noqa: PLR0913
             heads=heads,
             dim_head=dim_head,
             mlp_ratio=mlp_ratio,
-            bands=bands,
             band_groups=band_groups,
             dropout=dropout,
             emb_dropout=emb_dropout,
@@ -622,7 +616,6 @@ def __init__(  # noqa: PLR0913
             heads=decoder_heads,
             dim_head=decoder_dim_head,
             mlp_ratio=decoder_mlp_ratio,
-            bands=bands,
             band_groups=band_groups,
             dropout=decoder_dropout,
         )