Skip to content

Commit

Permalink
Drop torch requirements (#1015)
Browse files Browse the repository at this point in the history
* Remove torch requirements

* Fix format

---------

Co-authored-by: Carolin Benjamins <benjamins@tnt.uni-hannover.de>
  • Loading branch information
benjamc and Carolin Benjamins authored May 23, 2023
1 parent df6659f commit af60825
Show file tree
Hide file tree
Showing 4 changed files with 4 additions and 6 deletions.
3 changes: 2 additions & 1 deletion examples/4_advanced_optimizer/3_metadata_callback.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@
"""

import sys

from ConfigSpace import Configuration, ConfigurationSpace, Float

from smac import HyperparameterOptimizationFacade as HPOFacade
Expand Down Expand Up @@ -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,
Expand Down
3 changes: 0 additions & 3 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand All @@ -40,7 +38,6 @@ def read_file(filepath: str) -> str:
"pydocstyle",
"flake8",
"pre-commit",
*torch_requirements,
],
}

Expand Down
2 changes: 1 addition & 1 deletion tests/fixtures/scenario.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
2 changes: 1 addition & 1 deletion tests/test_initial_design/test_initial_design.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand All @@ -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

0 comments on commit af60825

Please sign in to comment.