From 7ba299c5521f3cf63d189352f8b0fc6d764ca7cd Mon Sep 17 00:00:00 2001 From: Collin Farquhar Date: Tue, 9 Apr 2024 12:22:12 -0600 Subject: [PATCH 1/2] import single qubit envs --- src/relaqs/environments/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/relaqs/environments/__init__.py b/src/relaqs/environments/__init__.py index e69de29b..11b93ba1 100644 --- a/src/relaqs/environments/__init__.py +++ b/src/relaqs/environments/__init__.py @@ -0,0 +1,2 @@ +from .single_qubit_env import SingleQubitEnv +from .noisy_single_qubit_env import NoisySingleQubitEnv From 7499005214260c6d79ef25b93d1726d10b85706d Mon Sep 17 00:00:00 2001 From: Collin Farquhar Date: Tue, 9 Apr 2024 12:22:29 -0600 Subject: [PATCH 2/2] use updated env class names --- scripts/hyperparameter_tuning/hyperparameter_tuning.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/hyperparameter_tuning/hyperparameter_tuning.py b/scripts/hyperparameter_tuning/hyperparameter_tuning.py index beca9d61..ab07b572 100644 --- a/scripts/hyperparameter_tuning/hyperparameter_tuning.py +++ b/scripts/hyperparameter_tuning/hyperparameter_tuning.py @@ -3,7 +3,7 @@ from ray import tune from ray.air import RunConfig from ray.rllib.algorithms.ddpg import DDPGConfig -from relaqs.environments.gate_synth_env_rllib_Haar import TwoQubitGateSynth, GateSynthEnvRLlibHaarNoisy +from relaqs.environments import SingleQubitEnv, NoisySingleQubitEnv from ray.tune.search.optuna import OptunaSearch from relaqs import RESULTS_DIR import datetime @@ -84,7 +84,7 @@ def objective(config): return results if __name__ == "__main__": - environment = GateSynthEnvRLlibHaarNoisy + environment = NoisySingleQubitEnv n_configurations = 25 n_training_iterations = 100 save = True