remove typo

MannLabs · May 29, 2024 · 610d158 · 610d158
1 parent 96921d5
commit 610d158
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sparcscore/ml/plmodels.py b/src/sparcscore/ml/plmodels.py
@@ -168,7 +168,7 @@ def test_step(self, batch, batch_idx):
 
 class RegressionModel(pl.LightningModule):
 
-    def __init__(self, model_type="VGG2_regression" **kwargs):
+    def __init__(self, model_type="VGG2_regression", **kwargs):
         super().__init__()
         self.save_hyperparameters()