Merge pull request #98 from automl/enhancement/dependency_versions

Add lower bound for all dependencies
automl · Dec 18, 2023 · 411f0ed · 411f0ed
2 parents ea8554c + b806b00
commit 411f0ed
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 20 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,8 @@
 # Version 1.1.4
 
+## Enhancements
+- Fix lower bounds of dependency versions.
+
 ## Bug-Fixes
 - Don't convert BOHB runs with status 'running' (consistent with SMAC).
 

diff --git a/examples/record/mnist_pytorch.py b/examples/record/mnist_pytorch.py
@@ -63,15 +63,14 @@ def __init__(self, activation="relu", learning_rate=1e-4, dropout_rate=0.1, batc
             ]
         )
 
-        self.accuracy = Accuracy()
+        self.accuracy = Accuracy(task="multiclass", num_classes=self.num_classes)
 
     def prepare_data(self):
         # download
         MNIST(self.data_dir, train=True, download=True)
         MNIST(self.data_dir, train=False, download=True)
 
     def setup(self, stage=None):
-
         # Assign train/val datasets for use in dataloaders
         if stage == "fit" or stage is None:
             mnist_full = MNIST(self.data_dir, train=True, transform=self.transform)
@@ -257,10 +256,9 @@ def get_configspace(seed):
 
                     # The model weights are trained
                     trainer = pl.Trainer(
-                        accelerator="gpu",
+                        accelerator="cpu",
                         devices=1,
                         num_sanity_val_steps=0,  # No validation sanity
-                        auto_scale_batch_size="power",
                         deterministic=True,
                         min_epochs=epochs,
                         max_epochs=epochs,

diff --git a/requirements.txt b/requirements.txt
@@ -1,12 +1,12 @@
-wheel
-setuptools
+wheel>=0.41.2
+setuptools==68.2.2
 absl-py>=1.0.0
 jsonlines>=3.0.0
 pandas>=1.3.4
 numpy>=1.22.2
 matplotlib>=3.5.1
 seaborn>=0.13.0
-pyyaml
+pyyaml>=6.0.1
 
 # AutoML packages
 ConfigSpace==0.6.1
@@ -24,4 +24,4 @@ rq>=1.10.1
 # Pinning might be removed for dash>2.3.0
 werkzeug==2.0.3
 
-pyPDPPartitioner
+pyPDPPartitioner>=0.1.8
diff --git a/setup.py b/setup.py
@@ -13,22 +13,22 @@ def read_file(file_name):
     "dev": [
         # Tests
         "pytest>=4.6",
-        "pytest-cov",
-        "pytest-xdist",
-        "pytest-timeout",
-        "mypy",
-        "isort",
-        "black",
-        "pydocstyle",
-        "pre-commit",
-        "flake8",
+        "pytest-cov>=4.1.0",
+        "pytest-xdist>=3.3.1",
+        "pytest-timeout>=2.2.0",
+        "mypy>=1.6.1",
+        "isort>=5.12.0",
+        "black>=23.11.0",
+        "pydocstyle>=6.3.0",
+        "pre-commit>=3.5.0",
+        "flake8>=6.1.0",
         # Docs
         "automl-sphinx-theme>=0.1.10",
     ],
     "examples": [
-        "torch",
-        "torchvision",
-        "pytorch-lightning",
+        "torch>=2.1.0",
+        "torchvision>=0.16.0",
+        "pytorch-lightning>=2.1.1",
     ],
 }