diff --git a/examples/4_advanced_optimizer/3_metadata_callback.py b/examples/4_advanced_optimizer/3_metadata_callback.py index 1f7078f94..ab35f2862 100644 --- a/examples/4_advanced_optimizer/3_metadata_callback.py +++ b/examples/4_advanced_optimizer/3_metadata_callback.py @@ -19,6 +19,7 @@ """ import sys + from ConfigSpace import Configuration, ConfigurationSpace, Float from smac import HyperparameterOptimizationFacade as HPOFacade @@ -65,7 +66,7 @@ def train(self, config: Configuration, seed: int = 0) -> float: branch="Name of Active Branch", commit="Commit Hash", command=" ".join(sys.argv), - additional_information="Some Additional Information" + additional_information="Some Additional Information", ) ], logging_level=999999, diff --git a/setup.py b/setup.py index 79249cd5f..40bf1b4d0 100644 --- a/setup.py +++ b/setup.py @@ -20,9 +20,7 @@ def read_file(filepath: str) -> str: return fh.read() -torch_requirements = ["torch>=1.9.0", "gpytorch>=1.5.0", "pyro-ppl>=1.7.0", "botorch>=0.5.0"] extras_require = { - "gpytorch": torch_requirements, "dev": [ "setuptools", "types-setuptools", @@ -40,7 +38,6 @@ def read_file(filepath: str) -> str: "pydocstyle", "flake8", "pre-commit", - *torch_requirements, ], } diff --git a/tests/fixtures/scenario.py b/tests/fixtures/scenario.py index 621193987..4830c68c6 100644 --- a/tests/fixtures/scenario.py +++ b/tests/fixtures/scenario.py @@ -51,7 +51,7 @@ def _make( instance_features=instance_features, min_budget=min_budget, max_budget=max_budget, - use_default_config=use_default_config + use_default_config=use_default_config, ) return _make diff --git a/tests/test_initial_design/test_initial_design.py b/tests/test_initial_design/test_initial_design.py index d12e503ff..ed08db0d5 100644 --- a/tests/test_initial_design/test_initial_design.py +++ b/tests/test_initial_design/test_initial_design.py @@ -122,6 +122,7 @@ def test_select_configurations(make_scenario, configspace_small): with pytest.raises(NotImplementedError): dc.select_configurations() + def test_include_default_config(make_scenario, configspace_small): scenario = make_scenario(configspace_small, use_default_config=True) @@ -133,4 +134,3 @@ def test_include_default_config(make_scenario, configspace_small): # if use_default_config is True, then the default config should be included in the additional_configs default_config = scenario.configspace.get_default_configuration() assert default_config in dc._additional_configs -