Merge pull request #51 from akataba/cf/hpt-update

Cf/hpt update
akataba · Apr 11, 2024 · 5ccb908 · 5ccb908
2 parents 9c4066d + 7499005
commit 5ccb908
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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

diff --git 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