From 863ea053620cca730308b81815fe572488e79b23 Mon Sep 17 00:00:00 2001
From: Erik Hagendorn <ehagendorn@gmail.com>
Date: Mon, 18 Dec 2023 09:58:40 -0500
Subject: [PATCH] out_dim is already being passed via train_args

---
 tsimcne/tsimcne.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/tsimcne/tsimcne.py b/tsimcne/tsimcne.py
index 7936b79..7da1411 100644
--- a/tsimcne/tsimcne.py
+++ b/tsimcne/tsimcne.py
@@ -475,7 +475,6 @@ def fit(self, X: torch.utils.data.Dataset | str):
                     model=model,
                     loss=p.loss,
                     metric=p.metric,
-                    out_dim=self.out_dim,
                     **train_kwargs,
                 )
             trainer = pl.Trainer(max_epochs=n_epochs, devices=1)