Repo flattening + adjustments

automl · Nov 30, 2023 · c48417a · c48417a
1 parent eec1222
commit c48417a
Show file tree

Hide file tree

Showing 106 changed files with 30 additions and 51 deletions.
diff --git a/src/neps/NOTICE → neps/NOTICE b/src/neps/NOTICE → neps/NOTICE
diff --git a/src/neps/__init__.py → neps/__init__.py b/src/neps/__init__.py → neps/__init__.py
diff --git a/src/neps/api.py → neps/api.py b/src/neps/api.py → neps/api.py
@@ -10,9 +10,7 @@
 
 import ConfigSpace as CS
 
-import metahyper
-from metahyper import instance_from_map
-
+from .metahyper import instance_from_map, metahyper_run
 from .optimizers import BaseOptimizer, SearcherMapping
 from .plot.tensorboard_eval import tblogger
 from .search_spaces.parameter import Parameter
@@ -258,7 +256,7 @@ def run(
             "skipped. Accurate continuation of runs can no longer be guaranteed!"
         )
 
-    metahyper.run(
+    metahyper_run(
         run_pipeline,
         searcher_instance,
         searcher_info,

diff --git a/neps/metahyper/__init__.py b/neps/metahyper/__init__.py
@@ -0,0 +1,2 @@
+from .api import ConfigResult, Sampler, read, metahyper_run
+from .utils import instance_from_map
diff --git a/src/metahyper/_locker.py → neps/metahyper/_locker.py b/src/metahyper/_locker.py → neps/metahyper/_locker.py
diff --git a/src/metahyper/api.py → neps/metahyper/api.py b/src/metahyper/api.py → neps/metahyper/api.py
@@ -413,7 +413,7 @@ def _evaluate_config(
     return result, {"time_end": time.time()}
 
 
-def run(
+def metahyper_run(
     evaluation_fn,
     sampler: Sampler,
     sampler_info: dict,

diff --git a/src/metahyper/example.py → neps/metahyper/example.py b/src/metahyper/example.py → neps/metahyper/example.py
diff --git a/src/metahyper/utils.py → neps/metahyper/utils.py b/src/metahyper/utils.py → neps/metahyper/utils.py
diff --git a/src/neps/optimizers/README.md → neps/optimizers/README.md b/src/neps/optimizers/README.md → neps/optimizers/README.md
diff --git a/src/neps/optimizers/__init__.py → neps/optimizers/__init__.py b/src/neps/optimizers/__init__.py → neps/optimizers/__init__.py
diff --git a/src/neps/optimizers/base_optimizer.py → neps/optimizers/base_optimizer.py b/src/neps/optimizers/base_optimizer.py → neps/optimizers/base_optimizer.py
@@ -5,9 +5,9 @@
 from copy import deepcopy
 from typing import Any
 
-import metahyper
-from metahyper.api import ConfigResult
+import neps.metahyper as metahyper
 
+from ..metahyper.api import ConfigResult
 from ..search_spaces.search_space import SearchSpace
 from ..utils.common import get_rnd_state, set_rnd_state
 from ..utils.result_utils import get_cost, get_learning_curve, get_loss

diff --git a/...ization/acquisition_functions/__init__.py → ...ization/acquisition_functions/__init__.py b/...ization/acquisition_functions/__init__.py → ...ization/acquisition_functions/__init__.py
diff --git a/...timization/acquisition_functions/_ehvi.py → ...timization/acquisition_functions/_ehvi.py b/...timization/acquisition_functions/_ehvi.py → ...timization/acquisition_functions/_ehvi.py
diff --git a/...acquisition_functions/base_acquisition.py → ...acquisition_functions/base_acquisition.py b/...acquisition_functions/base_acquisition.py → ...acquisition_functions/base_acquisition.py
diff --git a/...ion/acquisition_functions/cost_cooling.py → ...ion/acquisition_functions/cost_cooling.py b/...ion/acquisition_functions/cost_cooling.py → ...ion/acquisition_functions/cost_cooling.py
diff --git a/..._optimization/acquisition_functions/ei.py → ..._optimization/acquisition_functions/ei.py b/..._optimization/acquisition_functions/ei.py → ..._optimization/acquisition_functions/ei.py
diff --git a/...timization/acquisition_functions/mf_ei.py → ...timization/acquisition_functions/mf_ei.py b/...timization/acquisition_functions/mf_ei.py → ...timization/acquisition_functions/mf_ei.py
diff --git a/...n/acquisition_functions/prior_weighted.py → ...n/acquisition_functions/prior_weighted.py b/...n/acquisition_functions/prior_weighted.py → ...n/acquisition_functions/prior_weighted.py
diff --git a/...optimization/acquisition_functions/ucb.py → ...optimization/acquisition_functions/ucb.py b/...optimization/acquisition_functions/ucb.py → ...optimization/acquisition_functions/ucb.py
diff --git a/...mization/acquisition_samplers/__init__.py → ...mization/acquisition_samplers/__init__.py b/...mization/acquisition_samplers/__init__.py → ...mization/acquisition_samplers/__init__.py
diff --git a/.../acquisition_samplers/base_acq_sampler.py → .../acquisition_samplers/base_acq_sampler.py b/.../acquisition_samplers/base_acq_sampler.py → .../acquisition_samplers/base_acq_sampler.py
diff --git a/...acquisition_samplers/evolution_sampler.py → ...acquisition_samplers/evolution_sampler.py b/...acquisition_samplers/evolution_sampler.py → ...acquisition_samplers/evolution_sampler.py
diff --git a/...quisition_samplers/freeze_thaw_sampler.py → ...quisition_samplers/freeze_thaw_sampler.py b/...quisition_samplers/freeze_thaw_sampler.py → ...quisition_samplers/freeze_thaw_sampler.py
diff --git a/.../acquisition_samplers/mutation_sampler.py → .../acquisition_samplers/mutation_sampler.py b/.../acquisition_samplers/mutation_sampler.py → .../acquisition_samplers/mutation_sampler.py
diff --git a/...on/acquisition_samplers/random_sampler.py → ...on/acquisition_samplers/random_sampler.py b/...on/acquisition_samplers/random_sampler.py → ...on/acquisition_samplers/random_sampler.py
diff --git a/...ers/bayesian_optimization/cost_cooling.py → ...ers/bayesian_optimization/cost_cooling.py b/...ers/bayesian_optimization/cost_cooling.py → ...ers/bayesian_optimization/cost_cooling.py
@@ -2,8 +2,7 @@
 
 from typing import Any
 
-from metahyper import ConfigResult, instance_from_map
-
+from ...metahyper import ConfigResult, instance_from_map
 from ...optimizers.bayesian_optimization.acquisition_functions.cost_cooling import (
     CostCooler,
 )

diff --git a/...bayesian_optimization/kernels/__init__.py → ...bayesian_optimization/kernels/__init__.py b/...bayesian_optimization/kernels/__init__.py → ...bayesian_optimization/kernels/__init__.py
diff --git a/...n_optimization/kernels/combine_kernels.py → ...n_optimization/kernels/combine_kernels.py b/...n_optimization/kernels/combine_kernels.py → ...n_optimization/kernels/combine_kernels.py
diff --git a/...tion/kernels/combine_kernels_hierarchy.py → ...tion/kernels/combine_kernels_hierarchy.py b/...tion/kernels/combine_kernels_hierarchy.py → ...tion/kernels/combine_kernels_hierarchy.py
diff --git a/...bayesian_optimization/kernels/encoding.py → ...bayesian_optimization/kernels/encoding.py b/...bayesian_optimization/kernels/encoding.py → ...bayesian_optimization/kernels/encoding.py
diff --git a/...esian_optimization/kernels/get_kernels.py → ...esian_optimization/kernels/get_kernels.py b/...esian_optimization/kernels/get_kernels.py → ...esian_optimization/kernels/get_kernels.py
@@ -1,7 +1,6 @@
 from __future__ import annotations
 
-from metahyper import instance_from_map
-
+from ....metahyper import instance_from_map
 from ....search_spaces.architecture.core_graph_grammar import CoreGraphGrammar
 from ....search_spaces.hyperparameters.categorical import CategoricalParameter
 from ....search_spaces.hyperparameters.float import FloatParameter

diff --git a/.../kernels/grakel_replace/edge_histogram.py → .../kernels/grakel_replace/edge_histogram.py b/.../kernels/grakel_replace/edge_histogram.py → .../kernels/grakel_replace/edge_histogram.py
diff --git a/...imization/kernels/grakel_replace/utils.py → ...imization/kernels/grakel_replace/utils.py b/...imization/kernels/grakel_replace/utils.py → ...imization/kernels/grakel_replace/utils.py
diff --git a/...ernels/grakel_replace/vertex_histogram.py → ...ernels/grakel_replace/vertex_histogram.py b/...ernels/grakel_replace/vertex_histogram.py → ...ernels/grakel_replace/vertex_histogram.py
diff --git a/...rnels/grakel_replace/weisfeiler_lehman.py → ...rnels/grakel_replace/weisfeiler_lehman.py b/...rnels/grakel_replace/weisfeiler_lehman.py → ...rnels/grakel_replace/weisfeiler_lehman.py
diff --git a/...sian_optimization/kernels/graph_kernel.py → ...sian_optimization/kernels/graph_kernel.py b/...sian_optimization/kernels/graph_kernel.py → ...sian_optimization/kernels/graph_kernel.py
diff --git a/...rs/bayesian_optimization/kernels/utils.py → ...rs/bayesian_optimization/kernels/utils.py b/...rs/bayesian_optimization/kernels/utils.py → ...rs/bayesian_optimization/kernels/utils.py
diff --git a/...optimization/kernels/vectorial_kernels.py → ...optimization/kernels/vectorial_kernels.py b/...optimization/kernels/vectorial_kernels.py → ...optimization/kernels/vectorial_kernels.py
diff --git a/...n_optimization/kernels/weisfilerlehman.py → ...n_optimization/kernels/weisfilerlehman.py b/...n_optimization/kernels/weisfilerlehman.py → ...n_optimization/kernels/weisfilerlehman.py
diff --git a/...ptimizers/bayesian_optimization/mf_tpe.py → ...ptimizers/bayesian_optimization/mf_tpe.py b/...ptimizers/bayesian_optimization/mf_tpe.py → ...ptimizers/bayesian_optimization/mf_tpe.py
@@ -9,8 +9,7 @@
 from scipy.stats import spearmanr
 from typing_extensions import Literal
 
-from metahyper import ConfigResult, instance_from_map
-
+from ...metahyper import ConfigResult, instance_from_map
 from ...search_spaces import (
     CategoricalParameter,
     ConstantParameter,

diff --git a/.../bayesian_optimization/models/__init__.py → .../bayesian_optimization/models/__init__.py b/.../bayesian_optimization/models/__init__.py → .../bayesian_optimization/models/__init__.py
@@ -1,4 +1,4 @@
-from metahyper.utils import MissingDependencyError
+from neps.metahyper.utils import MissingDependencyError
 
 from .gp import ComprehensiveGP
 from .gp_hierarchy import ComprehensiveGPHierarchy

diff --git a/...rs/bayesian_optimization/models/deepGP.py → ...rs/bayesian_optimization/models/deepGP.py b/...rs/bayesian_optimization/models/deepGP.py → ...rs/bayesian_optimization/models/deepGP.py
diff --git a/...mizers/bayesian_optimization/models/gp.py → ...mizers/bayesian_optimization/models/gp.py b/...mizers/bayesian_optimization/models/gp.py → ...mizers/bayesian_optimization/models/gp.py
diff --git a/...esian_optimization/models/gp_hierarchy.py → ...esian_optimization/models/gp_hierarchy.py b/...esian_optimization/models/gp_hierarchy.py → ...esian_optimization/models/gp_hierarchy.py
diff --git a/...mizers/bayesian_optimization/optimizer.py → ...mizers/bayesian_optimization/optimizer.py b/...mizers/bayesian_optimization/optimizer.py → ...mizers/bayesian_optimization/optimizer.py
@@ -5,8 +5,7 @@
 
 from typing_extensions import Literal
 
-from metahyper import ConfigResult, instance_from_map
-
+from ...metahyper import ConfigResult, instance_from_map
 from ...search_spaces.hyperparameters.categorical import (
     CATEGORICAL_CONFIDENCE_SCORES,
     CategoricalParameter,

diff --git a/...ps/optimizers/default_searchers/asha.yaml → neps/optimizers/default_searchers/asha.yaml b/...ps/optimizers/default_searchers/asha.yaml → neps/optimizers/default_searchers/asha.yaml
diff --git a/...imizers/default_searchers/asha_prior.yaml → ...imizers/default_searchers/asha_prior.yaml b/...imizers/default_searchers/asha_prior.yaml → ...imizers/default_searchers/asha_prior.yaml
diff --git a/...ault_searchers/bayesian_optimization.yaml → ...ault_searchers/bayesian_optimization.yaml b/...ault_searchers/bayesian_optimization.yaml → ...ault_searchers/bayesian_optimization.yaml
diff --git a/...timizers/default_searchers/hyperband.yaml → ...timizers/default_searchers/hyperband.yaml b/...timizers/default_searchers/hyperband.yaml → ...timizers/default_searchers/hyperband.yaml
diff --git a/...optimizers/default_searchers/mobster.yaml → ...optimizers/default_searchers/mobster.yaml b/...optimizers/default_searchers/mobster.yaml → ...optimizers/default_searchers/mobster.yaml
diff --git a/...ps/optimizers/default_searchers/pibo.yaml → neps/optimizers/default_searchers/pibo.yaml b/...ps/optimizers/default_searchers/pibo.yaml → neps/optimizers/default_searchers/pibo.yaml
diff --git a/...timizers/default_searchers/priorband.yaml → ...timizers/default_searchers/priorband.yaml b/...timizers/default_searchers/priorband.yaml → ...timizers/default_searchers/priorband.yaml
diff --git a/...izers/default_searchers/priorband_bo.yaml → ...izers/default_searchers/priorband_bo.yaml b/...izers/default_searchers/priorband_bo.yaml → ...izers/default_searchers/priorband_bo.yaml
diff --git a/...zers/default_searchers/random_search.yaml → ...zers/default_searchers/random_search.yaml b/...zers/default_searchers/random_search.yaml → ...zers/default_searchers/random_search.yaml
diff --git a/...ault_searchers/regularized_evolution.yaml → ...ault_searchers/regularized_evolution.yaml b/...ault_searchers/regularized_evolution.yaml → ...ault_searchers/regularized_evolution.yaml
diff --git a/...default_searchers/successive_halving.yaml → ...default_searchers/successive_halving.yaml b/...default_searchers/successive_halving.yaml → ...default_searchers/successive_halving.yaml
diff --git a/...t_searchers/successive_halving_prior.yaml → ...t_searchers/successive_halving_prior.yaml b/...t_searchers/successive_halving_prior.yaml → ...t_searchers/successive_halving_prior.yaml
diff --git a/src/neps/optimizers/grid_search/optimizer.py → neps/optimizers/grid_search/optimizer.py b/src/neps/optimizers/grid_search/optimizer.py → neps/optimizers/grid_search/optimizer.py
@@ -2,8 +2,7 @@
 
 import random
 
-from metahyper.api import ConfigResult
-
+from ...metahyper import ConfigResult
 from ...search_spaces.search_space import SearchSpace
 from ..base_optimizer import BaseOptimizer
 

diff --git a/src/neps/optimizers/info.py → neps/optimizers/info.py b/src/neps/optimizers/info.py → neps/optimizers/info.py
diff --git a/src/neps/optimizers/multi_fidelity/_dyhpo.py → neps/optimizers/multi_fidelity/_dyhpo.py b/src/neps/optimizers/multi_fidelity/_dyhpo.py → neps/optimizers/multi_fidelity/_dyhpo.py
@@ -3,8 +3,7 @@
 
 import numpy as np
 
-from metahyper import ConfigResult
-
+from ...metahyper import ConfigResult
 from ...search_spaces.search_space import FloatParameter, IntegerParameter, SearchSpace
 from ..base_optimizer import BaseOptimizer
 from ..bayesian_optimization.acquisition_functions.base_acquisition import BaseAcquisition

diff --git a/src/neps/optimizers/multi_fidelity/dyhpo.py → neps/optimizers/multi_fidelity/dyhpo.py b/src/neps/optimizers/multi_fidelity/dyhpo.py → neps/optimizers/multi_fidelity/dyhpo.py
@@ -6,8 +6,7 @@
 
 import numpy as np
 
-from metahyper import ConfigResult, instance_from_map
-
+from ...metahyper import ConfigResult, instance_from_map
 from ...search_spaces.search_space import FloatParameter, IntegerParameter, SearchSpace
 from ..base_optimizer import BaseOptimizer
 from ..bayesian_optimization.acquisition_functions import AcquisitionMapping

diff --git a/...ps/optimizers/multi_fidelity/hyperband.py → neps/optimizers/multi_fidelity/hyperband.py b/...ps/optimizers/multi_fidelity/hyperband.py → neps/optimizers/multi_fidelity/hyperband.py
@@ -9,8 +9,7 @@
 import numpy as np
 from typing_extensions import Literal
 
-from metahyper import ConfigResult
-
+from ...metahyper import ConfigResult
 from ...search_spaces.search_space import SearchSpace
 from ..bayesian_optimization.acquisition_functions.base_acquisition import BaseAcquisition
 from ..bayesian_optimization.acquisition_samplers.base_acq_sampler import (

diff --git a/src/neps/optimizers/multi_fidelity/mf_bo.py → neps/optimizers/multi_fidelity/mf_bo.py b/src/neps/optimizers/multi_fidelity/mf_bo.py → neps/optimizers/multi_fidelity/mf_bo.py
@@ -7,8 +7,7 @@
 import pandas as pd
 import torch
 
-from metahyper import instance_from_map
-
+from ...metahyper import instance_from_map
 from ..bayesian_optimization.models import SurrogateModelMapping
 from ..multi_fidelity.utils import normalize_vectorize_config
 from ..multi_fidelity_prior.utils import calc_total_resources_spent, update_fidelity

diff --git a/...mizers/multi_fidelity/promotion_policy.py → ...mizers/multi_fidelity/promotion_policy.py b/...mizers/multi_fidelity/promotion_policy.py → ...mizers/multi_fidelity/promotion_policy.py
diff --git a/...imizers/multi_fidelity/sampling_policy.py → ...imizers/multi_fidelity/sampling_policy.py b/...imizers/multi_fidelity/sampling_policy.py → ...imizers/multi_fidelity/sampling_policy.py
@@ -10,8 +10,7 @@
 import pandas as pd
 import torch
 
-from metahyper import instance_from_map
-
+from ...metahyper import instance_from_map
 from ...search_spaces.search_space import SearchSpace
 from ..bayesian_optimization.acquisition_functions import AcquisitionMapping
 from ..bayesian_optimization.acquisition_functions.base_acquisition import BaseAcquisition

diff --git a/...zers/multi_fidelity/successive_halving.py → ...zers/multi_fidelity/successive_halving.py b/...zers/multi_fidelity/successive_halving.py → ...zers/multi_fidelity/successive_halving.py
@@ -10,8 +10,7 @@
 import pandas as pd
 from typing_extensions import Literal
 
-from metahyper import ConfigResult
-
+from ...metahyper import ConfigResult
 from ...search_spaces.hyperparameters.categorical import (
     CATEGORICAL_CONFIDENCE_SCORES,
     CategoricalParameter,

diff --git a/src/neps/optimizers/multi_fidelity/utils.py → neps/optimizers/multi_fidelity/utils.py b/src/neps/optimizers/multi_fidelity/utils.py → neps/optimizers/multi_fidelity/utils.py
diff --git a/...timizers/multi_fidelity_prior/__init__.py → ...timizers/multi_fidelity_prior/__init__.py b/...timizers/multi_fidelity_prior/__init__.py → ...timizers/multi_fidelity_prior/__init__.py
diff --git a/...s/multi_fidelity_prior/async_priorband.py → ...s/multi_fidelity_prior/async_priorband.py b/...s/multi_fidelity_prior/async_priorband.py → ...s/multi_fidelity_prior/async_priorband.py
@@ -5,8 +5,7 @@
 import numpy as np
 from typing_extensions import Literal
 
-from metahyper import ConfigResult
-
+from ...metahyper import ConfigResult
 from ...search_spaces.search_space import SearchSpace
 from ..bayesian_optimization.acquisition_functions.base_acquisition import BaseAcquisition
 from ..bayesian_optimization.acquisition_samplers.base_acq_sampler import (

diff --git a/...imizers/multi_fidelity_prior/priorband.py → ...imizers/multi_fidelity_prior/priorband.py b/...imizers/multi_fidelity_prior/priorband.py → ...imizers/multi_fidelity_prior/priorband.py
diff --git a/.../optimizers/multi_fidelity_prior/utils.py → .../optimizers/multi_fidelity_prior/utils.py b/.../optimizers/multi_fidelity_prior/utils.py → .../optimizers/multi_fidelity_prior/utils.py
diff --git a/..._knowledge_sources/prototype_optimizer.py → ..._knowledge_sources/prototype_optimizer.py b/..._knowledge_sources/prototype_optimizer.py → ..._knowledge_sources/prototype_optimizer.py
@@ -3,8 +3,7 @@
 import logging
 from typing import Any
 
-from metahyper.api import ConfigResult
-
+from ...metahyper import ConfigResult
 from ...search_spaces.search_space import SearchSpace
 from ...utils.data_loading import read_tasks_and_dev_stages_from_disk
 from .. import BaseOptimizer

diff --git a/...eps/optimizers/random_search/optimizer.py → neps/optimizers/random_search/optimizer.py b/...eps/optimizers/random_search/optimizer.py → neps/optimizers/random_search/optimizer.py
@@ -1,7 +1,6 @@
 from __future__ import annotations
 
-from metahyper.api import ConfigResult
-
+from ...metahyper import ConfigResult
 from ...search_spaces.search_space import SearchSpace
 from ..base_optimizer import BaseOptimizer
 

diff --git a/...mizers/regularized_evolution/optimizer.py → ...mizers/regularized_evolution/optimizer.py b/...mizers/regularized_evolution/optimizer.py → ...mizers/regularized_evolution/optimizer.py
diff --git a/src/neps/optimizers/utils.py → neps/optimizers/utils.py b/src/neps/optimizers/utils.py → neps/optimizers/utils.py
diff --git a/src/neps/plot/__init__.py → neps/plot/__init__.py b/src/neps/plot/__init__.py → neps/plot/__init__.py
diff --git a/src/neps/plot/__main__.py → neps/plot/__main__.py b/src/neps/plot/__main__.py → neps/plot/__main__.py
diff --git a/src/neps/plot/plot.py → neps/plot/plot.py b/src/neps/plot/plot.py → neps/plot/plot.py
diff --git a/src/neps/plot/plotting.py → neps/plot/plotting.py b/src/neps/plot/plotting.py → neps/plot/plotting.py
diff --git a/src/neps/plot/read_results.py → neps/plot/read_results.py b/src/neps/plot/read_results.py → neps/plot/read_results.py
diff --git a/src/neps/plot/tensorboard_eval.py → neps/plot/tensorboard_eval.py b/src/neps/plot/tensorboard_eval.py → neps/plot/tensorboard_eval.py
@@ -9,8 +9,7 @@
 from torch.utils.tensorboard import SummaryWriter
 from torch.utils.tensorboard.summary import hparams
 
-from metahyper.api import ConfigInRun
-
+from ..metahyper.api import ConfigInRun
 from ..status.status import get_summary_dict
 from ..utils.common import get_initial_directory
 

diff --git a/src/neps/search_spaces/__init__.py → neps/search_spaces/__init__.py b/src/neps/search_spaces/__init__.py → neps/search_spaces/__init__.py
diff --git a/src/neps/search_spaces/architecture/api.py → neps/search_spaces/architecture/api.py b/src/neps/search_spaces/architecture/api.py → neps/search_spaces/architecture/api.py
diff --git a/src/neps/search_spaces/architecture/cfg.py → neps/search_spaces/architecture/cfg.py b/src/neps/search_spaces/architecture/cfg.py → neps/search_spaces/architecture/cfg.py
diff --git a/...chitecture/cfg_variants/cfg_resolution.py → ...chitecture/cfg_variants/cfg_resolution.py b/...chitecture/cfg_variants/cfg_resolution.py → ...chitecture/cfg_variants/cfg_resolution.py
diff --git a/...hitecture/cfg_variants/constrained_cfg.py → ...hitecture/cfg_variants/constrained_cfg.py b/...hitecture/cfg_variants/constrained_cfg.py → ...hitecture/cfg_variants/constrained_cfg.py
diff --git a/...spaces/architecture/core_graph_grammar.py → ...spaces/architecture/core_graph_grammar.py b/...spaces/architecture/core_graph_grammar.py → ...spaces/architecture/core_graph_grammar.py
diff --git a/...s/search_spaces/architecture/crossover.py → neps/search_spaces/architecture/crossover.py b/...s/search_spaces/architecture/crossover.py → neps/search_spaces/architecture/crossover.py
diff --git a/src/neps/search_spaces/architecture/graph.py → neps/search_spaces/architecture/graph.py b/src/neps/search_spaces/architecture/graph.py → neps/search_spaces/architecture/graph.py
diff --git a/...arch_spaces/architecture/graph_grammar.py → ...arch_spaces/architecture/graph_grammar.py b/...arch_spaces/architecture/graph_grammar.py → ...arch_spaces/architecture/graph_grammar.py
diff --git a/...s/search_spaces/architecture/mutations.py → neps/search_spaces/architecture/mutations.py b/...s/search_spaces/architecture/mutations.py → neps/search_spaces/architecture/mutations.py
diff --git a/.../search_spaces/architecture/primitives.py → .../search_spaces/architecture/primitives.py b/.../search_spaces/architecture/primitives.py → .../search_spaces/architecture/primitives.py
diff --git a/.../search_spaces/architecture/topologies.py → .../search_spaces/architecture/topologies.py b/.../search_spaces/architecture/topologies.py → .../search_spaces/architecture/topologies.py
diff --git a/...rch_spaces/hyperparameters/categorical.py → ...rch_spaces/hyperparameters/categorical.py b/...rch_spaces/hyperparameters/categorical.py → ...rch_spaces/hyperparameters/categorical.py
diff --git a/...search_spaces/hyperparameters/constant.py → ...search_spaces/hyperparameters/constant.py b/...search_spaces/hyperparameters/constant.py → ...search_spaces/hyperparameters/constant.py
diff --git a/...ps/search_spaces/hyperparameters/float.py → neps/search_spaces/hyperparameters/float.py b/...ps/search_spaces/hyperparameters/float.py → neps/search_spaces/hyperparameters/float.py
diff --git a/.../search_spaces/hyperparameters/integer.py → .../search_spaces/hyperparameters/integer.py b/.../search_spaces/hyperparameters/integer.py → .../search_spaces/hyperparameters/integer.py
diff --git a/...earch_spaces/hyperparameters/numerical.py → ...earch_spaces/hyperparameters/numerical.py b/...earch_spaces/hyperparameters/numerical.py → ...earch_spaces/hyperparameters/numerical.py
diff --git a/src/neps/search_spaces/parameter.py → neps/search_spaces/parameter.py b/src/neps/search_spaces/parameter.py → neps/search_spaces/parameter.py
diff --git a/src/neps/search_spaces/search_space.py → neps/search_spaces/search_space.py b/src/neps/search_spaces/search_space.py → neps/search_spaces/search_space.py
diff --git a/src/neps/status/__main__.py → neps/status/__main__.py b/src/neps/status/__main__.py → neps/status/__main__.py
diff --git a/src/neps/status/status.py → neps/status/status.py b/src/neps/status/status.py → neps/status/status.py
@@ -6,10 +6,8 @@
 
 import pandas as pd
 
-from metahyper import read
-from metahyper._locker import Locker
-from metahyper.api import ConfigResult
-
+from ..metahyper import ConfigResult, read
+from ..metahyper._locker import Locker
 from ..search_spaces.search_space import SearchSpace
 from ..utils.result_utils import get_loss
 

diff --git a/src/neps/utils/common.py → neps/utils/common.py b/src/neps/utils/common.py → neps/utils/common.py
@@ -10,8 +10,7 @@
 import torch
 import yaml
 
-from metahyper.api import ConfigInRun
-
+from ..metahyper.api import ConfigInRun
 from ..optimizers.info import SearcherConfigs
 
 

diff --git a/src/neps/utils/data_loading.py → neps/utils/data_loading.py b/src/neps/utils/data_loading.py → neps/utils/data_loading.py
@@ -9,8 +9,7 @@
 import numpy as np
 import yaml
 
-import metahyper.api
-
+from ..metahyper import read
 from .result_utils import get_loss
 
 
@@ -40,7 +39,7 @@ def read_tasks_and_dev_stages_from_disk(
                 continue
             dev_id = get_id_from_path(dev_dir_path)
             # TODO: Perhaps use 2nd and 3rd argument as well
-            result, _, _ = metahyper.api.read(dev_dir_path)
+            result, _, _ = read(dev_dir_path)
             results[task_id][dev_id] = result
     return results
 
@@ -65,7 +64,7 @@ def read_user_prior_results_from_disk(path: str):
             continue
         # get name of the directory
         name = os.path.basename(prior_dir_path)
-        results[name], _, _ = metahyper.api.read(prior_dir_path)
+        results[name], _, _ = read(prior_dir_path)
     return results
 
 
@@ -163,7 +162,7 @@ def summarize_results(
             #  TOOD: only use IDs if provided
             final_results = results[final_task_id][final_dev_id]
         else:
-            final_results, _, _ = metahyper.api.read(seed_dir_path)
+            final_results, _, _ = read(seed_dir_path)
 
         # This part is copied from neps.status()
         best_loss = float("inf")

diff --git a/src/neps/utils/result_utils.py → neps/utils/result_utils.py b/src/neps/utils/result_utils.py → neps/utils/result_utils.py
diff --git a/src/metahyper/__init__.py b/src/metahyper/__init__.py
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		from .api import ConfigResult, Sampler, read, metahyper_run
		from .utils import instance_from_map