refactored to tuning_agent

Author: patrickhyw

env/integtest_pg_conn.py

@@ -5,7 +5,10 @@
 
 import yaml
 
-from env.integtest_util import ENV_INTEGTESTS_DBGYM_CONFIG_FPATH, get_integtest_workspace_path
+from env.integtest_util import (
+    ENV_INTEGTESTS_DBGYM_CONFIG_FPATH,
+    get_integtest_workspace_path,
+)
 from env.pg_conn import PostgresConn
 from util.pg import (
     DEFAULT_POSTGRES_PORT,

env/integtest_agent.py renamed to env/integtest_tuning_agent.py

@@ -1,8 +1,11 @@
 import subprocess
 import unittest
 
-from env.agent import TuningAgent
-from env.integtest_util import ENV_INTEGTESTS_DBGYM_CONFIG_FPATH, get_integtest_workspace_path
+from env.integtest_util import (
+    ENV_INTEGTESTS_DBGYM_CONFIG_FPATH,
+    get_integtest_workspace_path,
+)
+from env.tuning_agent import TuningAgent
 from util.workspace import DBGymConfig
 
 
@@ -25,4 +28,4 @@ def test_test(self) -> None:
 
 
 if __name__ == "__main__":
-    unittest.main()
+    unittest.main()

env/integtest_util.py

@@ -2,11 +2,10 @@
 
 import yaml
 
-
 ENV_INTEGTESTS_DBGYM_CONFIG_FPATH = Path("env/env_integtests_dbgym_config.yaml")
 
 
 def get_integtest_workspace_path() -> Path:
     with open(ENV_INTEGTESTS_DBGYM_CONFIG_FPATH) as f:
         return Path(yaml.safe_load(f)["dbgym_workspace_path"])
-    assert False
+    assert False

env/agent.py renamed to env/tuning_agent.py

@@ -2,7 +2,12 @@
 
 
 class DBMSConfig:
-    def __init__(self, indexes: list[str], sysknobs: dict[str, str], query_knobs: dict[str, list[str]]) -> None:
+    def __init__(
+        self,
+        indexes: list[str],
+        sysknobs: dict[str, str],
+        query_knobs: dict[str, list[str]],
+    ) -> None:
         self.indexes = indexes
         self.sysknobs = sysknobs
         self.query_knobs = query_knobs
@@ -20,4 +25,4 @@ def step(self) -> None:
 
     # Subclasses should override this function.
     def _step(self) -> DBMSConfig:
-        raise NotImplementedError
+        raise NotImplementedError