changed print to logger.info()

README.md

@@ -9,7 +9,7 @@ These steps were tested on a fresh repository clone, Ubuntu 22.04.
 
 ```
 # Setup dependencies.
-# You may want to create a Python virtual environment (e.g. with conda) before doing this.
+# You may want to create a Python 3.10 virtual environment (e.g. with conda) before doing this.
 ./dependency/install_dependencies.sh
 
 # Compile a custom fork of PostgreSQL, load TPC-H (SF 0.01), train the Proto-X agent, and tune.

manage/cli.py

@@ -8,7 +8,6 @@
 from itertools import chain
 import os
 
-from misc.utils import get_symlinks_path_from_workspace_path
 
 task_logger = logging.getLogger("task")
 task_logger.setLevel(logging.INFO)
@@ -207,8 +206,8 @@ def clean_workspace(dbgym_cfg: DBGymConfig, mode: str="safe", verbose=False) ->
     ending_num_files = _count_files_in_workspace(dbgym_cfg)
 
     if verbose:
-        print(f"Removed {starting_num_files - ending_num_files} out of {starting_num_files} files")
-        print(f"Workspace went from {starting_num_files - ending_num_files} to {starting_num_files}")
+        task_logger.info(f"Removed {starting_num_files - ending_num_files} out of {starting_num_files} files")
+        task_logger.info(f"Workspace went from {starting_num_files - ending_num_files} to {starting_num_files}")
 
 
 manage_group.add_command(manage_show)