./scripts/format.sh

cmu-db · Dec 30, 2024 · d4a3e99 · d4a3e99
1 parent 257ea05
commit d4a3e99
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 3 deletions.
diff --git a/benchmark/job/cli.py b/benchmark/job/cli.py
@@ -8,9 +8,9 @@
     get_workload_suffix,
     name_to_linkname,
 )
+from gymlib.shell import subprocess_run
 
 from benchmark.constants import DEFAULT_SCALE_FACTOR
-from util.shell import subprocess_run
 from util.workspace import DBGymWorkspace, fully_resolve_path
 
 JOB_TABLES_URL = "https://event.cwi.nl/da/job/imdb.tgz"

diff --git a/benchmark/tpch/cli.py b/benchmark/tpch/cli.py
@@ -10,10 +10,10 @@
     linkname_to_name,
     name_to_linkname,
 )
+from gymlib.shell import subprocess_run
 
 from benchmark.constants import DEFAULT_SCALE_FACTOR
 from benchmark.tpch.constants import DEFAULT_TPCH_SEED, NUM_TPCH_QUERIES
-from util.shell import subprocess_run
 from util.workspace import DBGymWorkspace, fully_resolve_path, is_fully_resolved
 
 TPCH_KIT_DIRNAME = "tpch-kit"

diff --git a/dbms/postgres/cli.py b/dbms/postgres/cli.py
@@ -16,6 +16,7 @@
     get_repo_symlink_path,
     linkname_to_name,
 )
+from gymlib.shell import subprocess_run
 
 from benchmark.constants import DEFAULT_SCALE_FACTOR
 from benchmark.job.load_info import JobLoadInfo
@@ -32,7 +33,6 @@
     sql_file_execute,
     sqlalchemy_conn_execute,
 )
-from util.shell import subprocess_run
 from util.workspace import (
     WORKSPACE_PATH_PLACEHOLDER,
     DBGymWorkspace,

diff --git a/gymlib_package/gymlib/__init__.py b/gymlib_package/gymlib/__init__.py
@@ -1,2 +1,5 @@
+# Everything at the "core" of gymlib will be imported directly (e.g. `from .[module] import *`).
+# Everything not as important will be only be imported as a module (e.g. `from . import [module]`).
+from . import shell
 from .magic import *
 from .symlinks_paths import *
diff --git a/util/shell.py → gymlib_package/gymlib/shell.py b/util/shell.py → gymlib_package/gymlib/shell.py