diff --git a/CHANGELOG.md b/CHANGELOG.md index 143cc91b..b3ef73be 100644 --- 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 index b50704cf..ea1772cb 100644 --- a/examples/record/mnist_pytorch.py +++ b/examples/record/mnist_pytorch.py @@ -63,7 +63,7 @@ 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 @@ -71,7 +71,6 @@ def prepare_data(self): 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 index 21764f6b..86ce889b 100644 --- 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 index 9fd44663..eab21c9b 100644 --- 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", ], }