fixing imports

akataba · Mar 8, 2024 · b49e16e · b49e16e
1 parent e28c8dd
commit b49e16e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/analysis/load_env_data.py b/analysis/load_env_data.py
@@ -3,7 +3,7 @@
 """
 
 from relaqs import RESULTS_DIR
-from relaqs.api import load_pickled_env_data
+from relaqs.api.utils import load_pickled_env_data
 
 data_path = RESULTS_DIR + '2024-01-24_11-37-15_X/env_data.pkl'
 

diff --git a/src/relaqs/api/__init__.py b/src/relaqs/api/__init__.py
@@ -1,4 +1,4 @@
 from .training import TrainRLLib
 from .callbacks import GateSynthesisCallbacks
 from .gates import Gate
-# from .utils import gate_fidelity, dm_fidelity, load_pickled_env_data
+