From 2c43944ec016e42c61bc476bbe761f8388c98324 Mon Sep 17 00:00:00 2001
From: Yizhou Chi <chiyizhou@fuzhi.ai>
Date: Wed, 4 Sep 2024 18:08:59 +0800
Subject: [PATCH] fix import

---
 expo/run_experiment.py | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/expo/run_experiment.py b/expo/run_experiment.py
index f8e58ce4f..8871c04a6 100644
--- a/expo/run_experiment.py
+++ b/expo/run_experiment.py
@@ -1,12 +1,10 @@
 import argparse
 import asyncio
 
-from expo.experimenter import (
-    AugExperimenter,
-    CustomExperimenter,
-    Experimenter,
-    MCTSExperimenter,
-)
+from expo.experimenter.aug import AugExperimenter
+from expo.experimenter.custom import CustomExperimenter
+from expo.experimenter.experimenter import Experimenter
+from expo.experimenter.mcts import MCTSExperimenter
 
 
 def get_args():