From cc83ad19326ad67dfe163bbe6b97791a6bd07b80 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?J=C3=A9r=C3=A9mie=20Gince?=
 <50332514+JeremieGince@users.noreply.github.com>
Date: Tue, 14 Oct 2025 22:54:39 -0400
Subject: [PATCH 1/4] Update README.md

---
 README.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index f768fe9..42ededa 100644
--- a/README.md
+++ b/README.md
@@ -93,7 +93,7 @@ class LinearNN(ClassificationModel):
     ):
         super().__init__(input_shape=input_shape, output_shape=output_shape, learning_rate=learning_rate, **kwargs)
         self.save_hyperparameters("learning_rate", "n_neurons")
-        self.nn = torch.Sequential(
+        self.nn = torch.nn.Sequential(
             torch.nn.Flatten(),
             torch.nn.LazyLinear(n_neurons), 
             torch.nn.ReLU(), 
@@ -107,7 +107,7 @@ class LinearNN(ClassificationModel):
     def output_size(self):
         return int(np.prod(self.output_shape))
 
-datamodule = DataModule.from_dataset_name("Digits2D")
+datamodule = DataModule.from_dataset_name("Digits2D", fold_id=0)
 automl_pipeline = AutoMLPipeline(model_cls=LinearNN, datamodule=datamodule)
 automl_pipeline.run()
 lt_pipeline, metrics = automl_pipeline.run_best_pipeline()

From 4862a76ad7c44db1091ee48c1b7ec965388c6321 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?J=C3=A9r=C3=A9mie=20Gince?=
 <50332514+JeremieGince@users.noreply.github.com>
Date: Tue, 14 Oct 2025 22:57:17 -0400
Subject: [PATCH 2/4] Update default epochs and time in AutoMLPipeline

Reduced DEFAULT_INNER_MAX_EPOCHS from 10,000 to 1,024 and DEFAULT_INNER_MAX_TIME from 90 hours to 1 hour in AutoMLPipeline. This likely aims to shorten training runs for faster experimentation or testing.
---
 src/matchcake_opt/tr_pipeline/automl_pipeline.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/matchcake_opt/tr_pipeline/automl_pipeline.py b/src/matchcake_opt/tr_pipeline/automl_pipeline.py
index 09c1832..e933431 100644
--- a/src/matchcake_opt/tr_pipeline/automl_pipeline.py
+++ b/src/matchcake_opt/tr_pipeline/automl_pipeline.py
@@ -19,8 +19,8 @@
 
 class AutoMLPipeline:
     DEFAULT_AUTOML_ITERATIONS = 32
-    DEFAULT_INNER_MAX_EPOCHS = 10_000
-    DEFAULT_INNER_MAX_TIME = "90:00:00:00"
+    DEFAULT_INNER_MAX_EPOCHS = 1024
+    DEFAULT_INNER_MAX_TIME = "00:01:00:00"
     DEFAULT_AUTOML_OVERWRITE_FIT = False
 
     @classmethod

From a2757b16886394ee81ce27b0992ebe825d1ac893 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?J=C3=A9r=C3=A9mie=20Gince?=
 <50332514+JeremieGince@users.noreply.github.com>
Date: Tue, 14 Oct 2025 23:11:42 -0400
Subject: [PATCH 3/4] Update README.md

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 42ededa..700666c 100644
--- a/README.md
+++ b/README.md
@@ -79,7 +79,7 @@ class LinearNN(ClassificationModel):
         RangeParameterConfig(
             name="n_neurons",
             parameter_type="int",
-            bounds=(4, 16),
+            bounds=(4, 2048),
         ),
     ]
 

From 6d0ed02912de50b3fe7d3d73bd1406a78114832e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?J=C3=A9r=C3=A9mie=20Gince?=
 <50332514+JeremieGince@users.noreply.github.com>
Date: Tue, 14 Oct 2025 23:37:02 -0400
Subject: [PATCH 4/4] Add instructions for CUDA (cu128) installation

Updated the README to include guidance on installing the package with CUDA support by using the `--extra cu128` flag.
---
 README.md | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/README.md b/README.md
index 700666c..843a9f5 100644
--- a/README.md
+++ b/README.md
@@ -35,6 +35,8 @@ With `uv` installed,
 uv add "git+https://github.com/MatchCake/MatchCake-Opt"
 ```
 
+To install the package with cu128 (CUDA), add `--extra cu128` to the installation commands above.
+
 
 ### For developers