Update pyproject.toml for PyTorch versions (#107)

neps/optimizers/bayesian_optimization/kernels/utils.py

@@ -33,11 +33,16 @@ def extract_configs(configs: list[SearchSpace]) -> Tuple[list, list]:
     """
     config_hps = [conf.get_normalized_hp_categories() for conf in configs]
     graphs = [hps["graphs"] for hps in config_hps]
-
-    _nested_graphs = np.array(graphs, dtype=object)
-    if _nested_graphs.ndim == 3:
-        graphs = _nested_graphs[:, :, 0].reshape(-1).tolist()
-
+    # Don't call np.array on structured objects
+    # https://github.com/numpy/numpy/issues/24546#issuecomment-1693913119
+    # _nested_graphs = np.array(graphs, dtype=object)
+    # if _nested_graphs.ndim == 3
+    #   graphs = _nested_graphs[:, :, 0].reshape(-1).tolist()
+    # Long hand way of doing the above
+    if (len(graphs) > 0 and isinstance(graphs[0], list)
+        and len(graphs[0]) > 0 and isinstance(graphs[0][0], list)):
+        res = [_list for list_of_list in graphs for _list in list_of_list]
+        graphs = res
     return graphs, config_hps
 
 

pyproject.toml

@@ -55,7 +55,8 @@ pandas = "^2"
 networkx = "^2.6.3"
 nltk = "^3.6.4"
 scipy = "^1"
-torch = ">=1.7.0,<=2.1, !=2.0.1, !=2.1.0" # fix from: https://stackoverflow.com/a/76647180
+# torch = ">=1.7.0,<=2.1, !=2.0.1, !=2.1.0" # fix from: https://stackoverflow.com/a/76647180
+torch = ">1.7.0,!=2.0.1, !=2.1.0"
 matplotlib = "^3"
 more-itertools = "*"
 portalocker = "^2"
@@ -70,7 +71,7 @@ pre-commit = "^3"
 mypy = "^1"
 pytest = "^7"
 types-PyYAML = "^6"
-torchvision = "<0.16.0"                               # Used in examples
+torchvision = ">=0.8.0"                               # Used in examples
 mkdocs-material = "*"
 mkdocs-autorefs = "*"
 mkdocs-gen-files = "*"

tests/test_examples.py

@@ -51,9 +51,9 @@ def no_logs_gte_error(caplog):
 def test_core_examples(example):
     if example.name == "analyse.py":
         # Run hyperparameters example to have something to analyse
-        runpy.run_path(core_examples_scripts[0], run_name="__main__")
+        runpy.run_path(str(core_examples_scripts[0]), run_name="__main__")
 
-    runpy.run_path(example, run_name="__main__")
+    runpy.run_path(str(example), run_name="__main__")
 
 
 @pytest.mark.ci_examples